0数学/找规律中等 LeetCode396. 旋转函数

描述

给定一个长度为 n 的整数数组 nums 。

假设 arrk 是数组 nums 顺时针旋转 k 个位置后的数组，我们定义 nums 的 旋转函数  F 为：

F(k) = 0 * arrk[0] + 1 * arrk[1] + … + (n - 1) * arrk[n - 1]
返回 F(0), F(1), …, F(n-1)中的最大值 。

生成的测试用例让答案符合 32 位 整数。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/rotate-function
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

找规律F(k) = F(k-1) + sum - n * nums[n-k]; sum是原数组所有元素之和。
评论区的题解：

class Solution {
    public int maxRotateFunction(int[] nums) {
        if (nums.length == 1) {
            return 0;
        }
        int base = 0;
        int n = nums.length;
        int sum = 0;
        for (int i = 0; i < n; i++) {
            base += i * nums[i];
            sum += nums[i];
        }
        int ans = base;
        for (int j = 1; j < n; j++) {
            int tmp = base + sum - n * nums[n-j];
            ans = Math.max(ans,tmp);
            base = tmp;
        }
        return ans;
    }
}