描述
给定一个长度为 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;
}
}