- 题目描述
-
给你一个整数数组 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发生变化。