leetcode刷题日记-三个无重叠子数组的最大和

• 题目描述

• 给你一个整数数组 nums 和一个整数 k ，找出三个长度为 k 、互不重叠、且 3 * k 项的和最大的子数组，并返回这三个子数组。
  以下标的数组形式返回结果，数组中的每一项分别指示每个子数组的起始位置（下标从 0 开始）。如果有多个结果，返回字典序最小的一个。

• 示例：

• 输入：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] 在字典序上更大。

• 解析：乍一看，不知道这个题目想干嘛，但是其实要考的东西很简单，这个题目就像之前的二数之和系列，一下给你做四数之和，你可能不会做，但是让你从二数、三数一步步做过来，你肯定就会了。这个题目就是这样，我们可以把题目先拆解一下，直接拆解为最简单的问题，如我们先求一个子数组的最大和，即给一个数组，如nums = [1,2,1,2,6,7,5,1], k = 2，求长度为2的子数组的最大和，并返回最大子数组第一个数对应的索引，这是不是就很简单了，就是一个简单的滑动窗口，判断条件是sum1>maxSum1，遍历n-k+1次，时间复杂度O(n)。然后进一步提高点难度，求两个无重叠子数组的最大和，那么我们就很容易的想到，用两个滑动窗口，第一个滑动窗口的最大值的判断条件是sum1>maxSum1，第二个滑动窗口的判断条件是maxSum12>maxSum1+sum2，变动时记录其索引变化。这是不是就很简单了，回到这个问题，就很容易发现，这是一个三个滑动窗口的求解问题，第三个滑动窗口的判断条件是maxSumTotal>maxSum2+sum3，变动时记录其索引变化，当然这里要注意，第一个滑动窗口的索引变化了，不一定会引起第一+第二滑动窗口和对应的索引变化，同理，maxSum12变了，maxSumTotal对应的三个索引不一定会发生变化，因此maxSum12和maxSumTotal对应的索引需要单独维护。
• 可能有点抽象，举个栗子，例如，nums = [1,2,2,3,6,7,5,1], k = 2
• 第一轮：sum1=3, maxSum1 = 3， maxSum1Index = 0; sum2=5 , maxSum12= 8， maxSum12Index = (0, 2);sum3=13 , maxSumTotal=21, maxSumTotalIndex=[0, 2, 4]
• 第二轮：sum1=4, maxSum1 = 4， maxSum1Index = 1; sum2=9 , maxSum12= 13， maxSum12Index = (1， 3); sum3=12 , maxSumTotal=25, maxSumTotalIndex=[1, 3, 5]
• 第三轮：sum1=5, maxSum1 = 5， maxSum1Index = 2; sum2=13 , maxSum12= 18， maxSum12Index = (2, 4); sum3=6 , maxSumTotal=25, maxSumTotalIndex=[1, 3, 5]
• 从上面的步骤可以看出，第三轮中，maxSum12Index发生了变化，但是maxSumTotalIndex不变
• 代码如下：

class Solution:
    def maxSumOfThreeSubarrays(self, nums: List[int], k: int) -> List[int]:
        """
        给你一个整数数组 nums 和一个整数 k ，找出三个长度为 k 、互不重叠、且 3      * k 项的和最大的子数组，并返回这三个子数组对应第一个元素的索引
        >>>self.maxSumOfThreeSubarrays([1,2,1,2,6,7,5,1], 2)
        >>>[0,3,5]
        """
        res = []
        sum1, maxSum1, maxSum1Index = 0, 0, 0
        sum2, maxSum2, maxSum2Index = 0, 0, ()
        sum3, maxSum3 = 0, 0
        for i in range(2*k, len(nums)):
            sum1 += nums[i-2*k]
            sum2 += nums[i-k]
            sum3 += nums[i]
            if i >= 3*k-1:
                if sum1 > maxSum1:
                    maxSum1 = sum1
                    maxSum1Index = i - 3*k + 1
                if maxSum1 + sum2 > maxSum2:
                    maxSum2 = maxSum1 + sum2
                    maxSum2Index = (maxSum1Index, i - 2*k + 1)
                if maxSum2 + sum3 > maxSum3:
                    maxSum3 = maxSum2 + sum3
                    res = [*maxSum2Index, i - k +1]
                print(sum1, maxSum1, maxSum1Index)
                print(sum2, maxSum2, maxSum2Index)
                print(sum3, maxSum3, res)
                sum1 -= nums[i - k*3 + 1]
                sum2 -= nums[i - k*2 + 1]
                sum3 -= nums[i - k + 1]
        return res

• *maxSum2Index的意思表示把元组maxSum2Index拆分为独立元素，可以简单理解为将元组中的元素以值传递的方式传递到列表res中，元组发生变化，不会引起列表res发生变化。