给定一个由整数数组 A 表示的环形数组 C,求 C 的非空子数组的最大可能和。
在此处,环形数组意味着数组的末端将会与开头相连呈环状。(形式上,当0 <= i < A.length 时 C[i] = A[i],且当 i >= 0 时 C[i+A.length] = C[i])
此外,子数组最多只能包含固定缓冲区 A 中的每个元素一次。(形式上,对于子数组 C[i], C[i+1], …, C[j],不存在 i <= k1, k2 <= j 其中 k1 % A.length = k2 % A.length)
示例 1:
输入:[1,-2,3,-2]
输出:3
解释:从子数组 [3] 得到最大和 3
示例 2:
输入:[5,-3,5]
输出:10
解释:从子数组 [5,5] 得到最大和 5 + 5 = 10
示例 3:
输入:[3,-1,2,-1]
输出:4
解释:从子数组 [2,-1,3] 得到最大和 2 + (-1) + 3 = 4
示例 4:
输入:[3,-2,2,-3]
输出:3
解释:从子数组 [3] 和 [3,-2,2] 都可以得到最大和 3
示例 5:
输入:[-2,-3,-1]
输出:-1
解释:从子数组 [-1] 得到最大和 -1
提示:
-30000 <= A[i] <= 30000
1 <= A.length <= 30000
相关标签
队列
数组
分治
动态规划
单调队列
这道题难一些的地方在于最大值在环形数组之间。动态规划主要问题是找出子状态和原来状态之间的关系:让我们对于左边的和:0~i,即从nums[0]累加到nums[i],其环形最大值位于[k, len-1],其中k>i。
以上思路的话,我们对于每个位置i,都会有一次循环找出这个最大值K,这样一来时间复杂度就比较高,那么我们可以先将循环中找的这个K用动态规划的方式得到[k, len-1]的最大值即可。
#include <bits/stdc++.h>
class Solution {
public:
int maxSubarraySumCircular(vector<int>& nums) {
int len = nums.size();
if(len == 1){
return nums[0];
}
int ans = nums[0];
/*第一种情况:最大值不在环形数组当中*/
int cur = nums[0];
for(int i=1; i<len; i++){
cur = nums[i] + max(cur, 0);
ans = max(ans, cur);
}
/*第二种情况: 最大值在环形数组中*/
//第二正方法: 以左边为基准
int leftSum[len], leftMax[len];
memset(leftSum, 0, sizeof(leftSum));
memset(leftMax, 0, sizeof(leftMax));
leftSum[0] = nums[0];
for(int i=1; i<len; i++){
leftSum[i] = leftSum[i-1] + nums[i];
}
leftMax[0] = nums[0];
for(int i=1; i<len; i++){
leftMax[i] = max(leftMax[i-1], leftSum[i]);
}
int rightSum = 0;
for(int i=len-1; i>=1; i--){
rightSum += nums[i];
ans = max(ans, rightSum+leftMax[i-1]);
}
return ans;
}
};