LeetCode-396. 旋转函数【动态规划,思路清晰,错位相减法,迭代】
2023-09-14 09:01:27 时间
题目描述:
给定一个长度为 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 位 整数。
示例 1:
输入: nums = [4,3,2,6]
输出: 26
解释:
F(0) = (0 * 4) + (1 * 3) + (2 * 2) + (3 * 6) = 0 + 3 + 4 + 18 = 25
F(1) = (0 * 6) + (1 * 4) + (2 * 3) + (3 * 2) = 0 + 4 + 6 + 6 = 16
F(2) = (0 * 2) + (1 * 6) + (2 * 4) + (3 * 3) = 0 + 6 + 8 + 9 = 23
F(3) = (0 * 3) + (1 * 2) + (2 * 6) + (3 * 4) = 0 + 2 + 12 + 12 = 26
所以 F(0), F(1), F(2), F(3) 中的最大值是 F(3) = 26 。
示例 2:
输入: nums = [100]
输出: 0
提示:
n == nums.length
1 <= n <= 105
-100 <= nums[i] <= 100
解题思路一:这种题一般不可暴力,需要找出F(n)与F(n-1)之间的关系,利用F(n-1)来求F(n),用数组存储,以空间换时间。公式具体过程看注释。
class Solution {
public:
//F(n)=F(n-1)+sum(nums)-(n)*arr[n-n]
//F(i)=F(i-1)+sum(nums)-(n)*arr[n-i]
//F(1)=25+15-4*6=16
//F(2)=16+15-4*2=23
//F(3)=23+15-4*3=26
//F(n)用数组存,sum固定,n*arr[n-i]
int maxRotateFunction(vector<int>& nums) {
int n=nums.size();
int sum=accumulate(nums.begin(),nums.end(),0);
vector<int> F(n);
int F0=0;
for(int i=0;i<n;++i){
F0+=i*nums[i];
}
F[0]=F0;//先算出F[0]
for(int i=1;i<n;++i){
F[i]=F[i-1]+sum-n*nums[n-i];
}
int ans=INT_MIN;
for(int i=0;i<n;++i){
ans=max(ans,F[i]);
}
return ans;
}
};
解题思路二:我们可以不用数组存储中间值,利用ans存储我们需要的最大值即可
用F[0]求之后的值
class Solution {
public:
int maxRotateFunction(vector<int>& nums) {
int n=nums.size();
int sum=accumulate(nums.begin(),nums.end(),0);
int F0=0;
for(int i=0;i<n;++i){//先算出F[0]
F0+=i*nums[i];
}
int ans=F0;
for(int i=n-1;i>=1;--i){
//F[i]=F[i-1]+sum-n*nums[n-i];
F0+=sum-n*nums[i];
ans=max(F0,ans);
}
return ans;
}
};
复杂度分析
时间复杂度:O(n),其中 n 是数组 nums 的长度。计算 numSum 和第一个 f 消耗 O(n) 时间,后续迭代 n−1 次 f消耗 O(n) 时间。
空间复杂度:O(1)。仅使用常数空间。
相关文章
- ☆打卡算法☆LeetCode 203. 移除链表元素 算法解析
- 拼写单词(leetcode 1160)
- LeetCode笔记:Biweekly Contest 83
- LeetCode笔记:Weekly Contest 303
- LeetCode周赛303,又见手速场……
- leetcode-91解码方法(动态规划|记忆化搜索)[通俗易懂]
- LeetCode刷题_LeetCode刷题手册
- leetcode刷题(131)——背包问题理解
- LeetCode 617. 合并二叉树
- leetcode-29. 两数相除
- 前端工程师leetcode算法面试必备-二分搜索算法(上)
- leetcode最长回文子串_最长回文子串算法
- js刷leetcode动态规划(图文视频讲解)
- LeetCode-77-组合
- LeetCode-416-分割等和子集
- 【Leetcode-链表强训】
- 【算法】动态规划 ⑦ ( LeetCode 55. 跳跃游戏 | 算法分析 | 代码示例 )
- 「动态规划」LeetCode 70(爬楼梯)