Leetcode 689. 三个无重叠子数组的最大和 C++

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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。