LeetCode 396. 旋转函数

题目链接：旋转函数

题解标签：数学、动态规划

思路：

        F(0) = 0*nums[0] + 1*nums[1] + ... + (n-2)*nums[n-2] + (n-1)*nums[n-1]

        F(1) = 0*nums[n-1] + 1*nums[0] + 2*nums[1] + ... + (n-1)*nums[n-2]

        很容易发现，红色部分差值为：nums[0] + nums[1] + ... + nums[n-2]，因此对于整数i，我们可以快速通过F(i)计算出F(i+1)。

        假设数组所有元素之和为sum，易得F(1) = F(0) + sum - n*nums[n-1]，

        同理可得F(2) = F(1) + sum - n*nums[n-2]，

        因此DP公式为：F(i) = F(i-1) + sum - n*nums[n-i]

代码：

class Solution {
        public int maxRotateFunction(int[] nums) {
            // sum为数组元素之和、now当前函数F的值
            int sum = 0, now = 0, len = nums.length;
            for (int i = 0; i < len; i++) {
                sum += nums[i];
                now += i * nums[i];
            }
            int ans = now;
            for (int i = 1; i < len; i++) {
                now = now + sum - len * nums[len - i];
                ans = Math.max(ans, now);
            }
            return ans;
        }
    }

时间复杂度：O(n)

运行结果：
            执行耗时:3 ms,击败了100.00% 的Java用户
            内存消耗:53.2 MB,击败了45.73% 的Java用户