给你一个整数数组 arr 和一个整数 k。
首先,我们要对该数组进行修改,即把原数组 arr 重复 k 次。
举个例子,如果 arr = [1, 2] 且 k = 3,那么修改后的数组就是 [1, 2, 1, 2, 1, 2]。
然后,请你返回修改后的数组中的最大的子数组之和。
注意,子数组长度可以是 0,在这种情况下它的总和也是 0。
由于 结果可能会很大,所以需要 模(mod) 10^9 + 7 后再返回。
示例 1:
输入:arr = [1,2], k = 3
输出:9
示例 2:
输入:arr = [1,-2,1], k = 5
输出:2
示例 3:
输入:arr = [-1,-2], k = 7
输出:0
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/k-concatenation-maximum-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:和剑指 Offer 42. 连续子数组的最大和的思路类似。时间复杂度上应该是O(N),但是不知道为什么会超时。…
定义状态:
f[i]
为到i
位置为止的最大子序和
class Solution {
public:
int kConcatenationMaxSum(vector<int>& arr, int k) {
vector<int>nums;
long long mod = 1e9 + 7;
int n = arr.size();
for(int i = 0;i < n*k;++i)
{
nums.emplace_back(arr[i % n]);
}
int res = INT_MIN;
vector<int>f(n*k,0);
f[0] = nums[0];
res = max(res,f[0]);
for(int i = 1;i < n*k;++i)
{
if(f[i-1]>0)
{
f[i] = f[i-1] + nums[i];
}
else
{
f[i] = nums[i];
}
res = max(res,f[i]);
}
return res < 0 ? 0:res%mod;
}
};