《LeetCode之每日一题》:229.三个无重叠子数组的最大和

题目链接： 三个无重叠子数组的最大和

有关题目

给你一个整数数组 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 * 10^4
1 <= nums[i] < 216
1 <= k <= floor(nums.length / 3)

题解

法一：滑动窗口
参考官方题解

class Solution {
public:
    vector<int> maxSumOfThreeSubarrays(vector<int>& nums, int k) {
        vector<int> ans;
        int sum1 = 0, maxSum1 = 0, maxSumIdx = 0;//maxSumIdx存放的为更新之后的sum1的最大值的下标
        int sum2 = 0, maxSum12 = 0, maxSumIdx1 = 0, maxSumIdx2 = 0;//maxSumIdx1, maxSumIdx2 中分别存放的为前两个子数组起始位置
        int sum3 = 0, maxTotal = 0;

        for (int i = 2 * k; i < nums.size(); i++)
        {
            sum1 += nums[i - 2 * k];
            sum2 += nums[i - k];
            sum3 += nums[i];

            if (i >= 3 * k - 1)
            {
                if (sum1 > maxSum1)//更新sum1及第一个子数组对应的下标
                {
                    maxSum1 = sum1;
                    maxSumIdx = i - 3 * k + 1;
                }

                if (sum2 + maxSum1 > maxSum12)//更新maxSum12及前两个子数组对应的下标
                {
                    maxSum12 = sum2 + maxSum1;
                    maxSumIdx1 = maxSumIdx;
                    maxSumIdx2 = i - 2 * k + 1;
                }

                if (sum3 + maxSum12 > maxTotal)//更新maxTotal及前三个子数组对应的下标
                {
                    maxTotal = sum3 + maxSum12;
                    ans = {maxSumIdx1, maxSumIdx2, i - k + 1};
                }

                //维护窗口的大小
                sum1 -= nums[i - 3 * k + 1];
                sum2 -= nums[i - 2 * k + 1];
                sum3 -= nums[i - k + 1];
            }
        }
        return ans;
    }
};