leetcode 918. 环形子数组的最大和 medium
题目描述:
给定一个由整数数组 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
解题思路:
环形数组,我们可以把原来的数组Adouble一下,这个新数组的所有长度小于等于n的子数组就包含了所有情况。比如
原来数组是 [1,-2,3,-2],新数组是[1,-2,3,-2,1,-2,3,-2] 。 而所有子数组的和我们可以用前缀和数组来快速得到,而以下标i为尾的最大子数组和,就是在i的左侧,并且距离小于n的最小前缀和,而这样一个区间的最小,我们可以用滑动窗口的方法来做。
代码:
//
class Solution {
public:
int maxSubarraySumCircular(vector<int>& A) {
int n=A.size();
vector<int> vec; // 把数组A double一下,这样就能表示所有的子数组
for(int i=0;i<2;++i)
for(int j=0;j<n;++j)
vec.push_back(A[j]);
//计算前缀和数组
vector<int> prefix(2*n+1);
for(int i=1;i<prefix.size();++i)
prefix[i]= prefix[i-1]+vec[i-1];
deque<int> deq;
deq.push_front(0); // 初始化
int res=INT_MIN;
for(int i=1;i<prefix.size();++i){
// 这里必须先减数,因为子数组不能为空
//因为deq里存的是前缀和的下标,而 s5-s2 = a2+a3+a4 是3个数的下标,所以这里是 i-n
if(deq.size() && deq.front()<i-n)
deq.pop_front();
res=max(res,prefix[i]-prefix[deq.front()]);
while(deq.size() && prefix[deq.back()]>=prefix[i])
deq.pop_back();
deq.push_back(i);
}
return res;
}
};