leetcode 689. 三个无重叠子数组的最大和——(每日一难day26)

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

给你一个整数数组 nums 和一个整数 k ，找出三个长度为 k 、互不重叠、且全部数字和（3 * k 项）最大的子数组，并返回这三个子数组。

以下标的数组形式返回结果，数组中的每一项分别指示每个子数组的起始位置（下标从 0 开始）。如果有多个结果，返回字典序最小的一个。

示例 1：

输入：nums = [1,2,1,2,6,7,5,1], k = 2
输出：[0,3,5]
解释：子数组 [1, 2], [2, 6], [7, 5] 对应的起始下标为 [0, 3, 5]。 也可以取 [2, 1], 但是结果 [1, 3, 5] 在字典序上更大。

示例 2：

输入：nums = [1,2,1,2,1,2,1,2,1], k = 2
输出：[0,2,4]

提示：

1 <= nums.length <= 2 * 1e4
1<= nums[i] < 216 1 <= k <=
floor(nums.length / 3)

解析

• 一个固定窗口的最大值应该怎样维护？固定大小窗口，从左往右滑动即可，遇到最大值更新窗口开始位置坐标。
• 两个窗口之和的最大值应该怎么维护？是不是可以枚举第二个窗口的位置。
• 假设第二个窗口起始位置为 i ，那么两个窗口的最大和应该是什么情况？是不是应该为i之前所有窗口的最大值加上 i 窗口的值就是以i起点的两个窗口之和，然后从枚举的所有i中找出最大值。显然复杂度为O（N*N）
• 怎么做可以降低复杂度呢？我们可以同时维护两个窗口，起始位置分别为[0-k-1],[k,2*k-1],同时往右移动，只要第一个窗口的最大值改变，我们就记录第一个窗口的最大值maxsum1（是不是就相当于记录了i之前的窗口最大值）和对应窗口下标，但是只有两个窗口之和发生改变时，整体下标才会更新，因为最终我们要找的是最大值。
• 三个窗口应该怎么做呢？同样的方法，可以枚举第三个窗口的位置 i，然后在移动过程中维护的前两个窗口的最大值maxsum2，前两个窗口的维护和两个窗口之和最大值问题是相同的。可以把前两个窗口看作一个整体转化为两个窗口的问题，三个之和发生改变时才会更新答案，如果前两个窗口最大值发生改变就将，前两个窗口的最大值更新，并记录下标 结合上一步两个窗口理解。
• 需要注意的是，可能存在第一个窗口值改变，但是前两个窗口总体值不发生改变（idx[2]不会更新），但是由于第三个窗口值过大，导致答案更新，此时也会更新第一个窗口位置，我们知道，如果前两个值不发生改变的话第一，二窗口下标不应该改变，因此只有前两个之和发生改变时才会同时更新第一个窗口下标。
  例如：2 5 2 9，k=1
  第一个窗口变成了5，但是前两个窗口和并没有变化，总体值变大此时更新idx[1]=1,maxsum2并没有改变，idx[2]并不会改变，因此idx[2]=[1]两者冲突。而第三个窗口始终不会和第二个重合的，因此不用同样处理idx[2]。

code

class Solution {
public:
    vector<int> maxSumOfThreeSubarrays(vector<int>& nums, int k) {
        int n =nums.size();
        int maxsum1=0,maxsum2=0,maxsum3=0;
        int sum1=0,sum2=0,sum3=0;
        int idx[4];
        vector<int> ans;
        ans={0,k,2*k};
        for(int i=2*k;i<n;i++){
            sum1+=nums[i-2*k];
            sum2+=nums[i-k];
            sum3+=nums[i];
            if(i>=3*k-1){
                if(sum1>maxsum1){
                    idx[0]=i-3*k+1;
                    maxsum1=sum1;
                }
                if(maxsum1+sum2>maxsum2){
                    idx[1]=idx[0];
                    idx[2]=i-2*k+1;
                    maxsum2=maxsum1+sum2;
                }
                if(maxsum2+sum3>maxsum3){
                    idx[3]=i-k+1;
                    maxsum3=maxsum2+sum3;
                    ans={idx[1],idx[2],idx[3]};
                }
                sum1-=nums[i-3*k+1];
                sum2-=nums[i-2*k+1];
                sum3-=nums[i-k+1];
            }
            
        }
        return ans;
    }
};