Leetcode 689. 三个无重叠子数组的最大和
题目
给定数组 nums 由正整数组成,找到三个互不重叠的子数组的最大和。
每个子数组的长度为k,我们要使这3*k个项的和最大化。
返回每个区间起始索引的列表(索引从 0 开始)。如果有多个结果,返回字典序最小的一个。
示例:
输入: [1,2,1,2,6,7,5,1], 2
输出: [0, 3, 5]
解释: 子数组 [1, 2], [2, 6], [7, 5] 对应的起始索引为 [0, 3, 5]。
我们也可以取 [2, 1], 但是结果 [1, 3, 5] 在字典序上更大。
注意:
- nums.length的范围在[1, 20000]之间。
- nums[i]的范围在[1, 65535]之间。
- k的范围在[1,floor(nums.length / 3)]之间。
题解
动态规划
我们先用滑动窗口求出大小为k的窗口的子数组和window
我们的目标就成了找出三个无重叠的窗口的和最大。
我们再用两个动态规划,dpL[i]、dpR[i]分别第i个窗口左边最大的窗口(包括第i个窗口)和,第i个窗口右边最大的窗口(包括第i个窗口)和。
- 此时的转移方程很好考虑,即考虑其左、本身;其右、本身
- dpL[i] = max(dpL[i-1],window[i]);
- dpR[size-i-1] = max(dpR[size-i],window[size-i-1]);
之后我们遍历窗口,将其作为中间的窗口,那么左边可以选择0 ~ i-k的窗口;右边可以选择i+k ~ window.size()-1的窗口。我们找到dpL[i-k]+window[i]+dpR[i+k]最大值即可,其中第i个窗口便是中间的窗口
最后我们根据dpL[m-k]、dpR[m+k]找到确定最先出现的左窗口、右窗口
详细过程见代码
代码
vector<int> maxSumOfThreeSubarrays(vector<int>& nums, int k) {
vector<int> window;
int n = nums.size();
int i=0,j=0,sum=0;
while(i <= n-k){ //求出0~n-k分别作为起始元素的大小为k的子数组和
while(j-i != k){
sum += nums[j];
j++;
}
window.push_back(sum);
sum -= nums[i];
i++;
}
int size = window.size();
vector<int> dpL(size,0),dpR(size,0);
dpL[0] = window[0];
dpR[size-1] = window[size-1];
for(i=1; i<size; i++){
dpL[i] = max(dpL[i-1],window[i]); //0~i窗口最大值
dpR[size-i-1] = max(dpR[size-i],window[size-i-1]); //i~size-1窗口最大值
}
int sumMax = 0;
int l,m,r;
for(i=k; i<size-k; i++){ //确定中间的窗口
if(dpL[i-k]+window[i]+dpR[i+k] > sumMax){
m = i;
sumMax = dpL[i-k]+window[i]+dpR[i+k];
}
}
for(i=0; i<m; i++) //找到最先出现的左窗口
if(dpL[m-k]==window[i]){
l = i;
break;
}
for(i=m+k; i<size; i++) //找到最先出现的右窗口
if(dpR[m+k]==window[i]){
r = i;
break;
}
return {l,m,r};
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-sum-of-3-non-overlapping-subarrays
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。