2021/12/08每日一题——三个无重叠子数组的最大和

题目

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

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

分析

利用滑动窗口：三个窗口的大小均为k，范围分别为[0, k-1]、[k，2k-1]、[2k，3k-1]，保证他们不重叠。
求每个窗口的最大和，即能求得三个窗口和的最大值。

代码

class Solution {
    public int[] maxSumOfThreeSubarrays(int[] nums, int k) {
        int[] ans = new int [3]; //存储结果下角标
        int sum1 = 0, maxSum1 = 0, index1 = 0; //窗口1的和、最大和
        int sum2 = 0, maxSum2 = 0, index2 = 0;//窗口2的和、两个窗口的最大和
        int sum3 = 0, maxSum3 = 0, index3 = 0;//窗口3的和、三个窗口的最大和

        for(int i=2*k;i<nums.length;i++){
            sum1 += nums[i-2*k]; //窗口1加入值
            sum2 += nums[i-k];//窗口2加入值
            sum3 += nums[i];//窗口3加入值

            //窗口中达到k个值，更新最大和
            if(i >= 3*k-1){
                if(sum1 > maxSum1){
                    maxSum1 = sum1;
                    index1 = i- 3 * k + 1;
                }
                if(sum2 + maxSum1 > maxSum2){
                    maxSum2 = sum2 + maxSum1;
                    index2 = index1;
                    index3 = i - 2 * k + 1;
                }
                if(sum3 + maxSum2 > maxSum3){
                    maxSum3 = sum3 + maxSum2; //找到比上一个和更大值，更新位置
                    ans[0] = index2;
                    ans[1] = index3;
                    ans[2] = i - k + 1;
                }
                //窗口向前滑动，减去最左边的值 
                sum1 -= nums[i - k * 3 + 1];
                sum2 -= nums[i - k * 2 + 1];
                sum3 -= nums[i - k + 1];
            }
        }
        return ans;
    }
}