目录
- 题目要求
- 示例
- 题解
-
- 嵌套多个滑动窗口
题目要求 给你一个整数数组 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:
【#|Leetcode每日一题——三个无重叠子数组的最大和】输入:nums = [1,2,1,2,1,2,1,2,1], k = 2 输出:[0,2,4]题解 嵌套多个滑动窗口
class Solution:
def maxSumOfThreeSubarrays(self, nums: List[int], k: int) -> List[int]:
ans = []
sum1, maxSum1, maxSum1Idx = 0, 0, 0
sum2, maxSum12, maxSum12Idx = 0, 0, ()
sum3, maxTotal = 0, 0
for i in range(k * 2, len(nums)):
sum1 += nums[i - k * 2]
sum2 += nums[i - k]
sum3 += nums[i]
if i >= k * 3 - 1:
if sum1 > maxSum1:
maxSum1 = sum1
maxSum1Idx = i - k * 3 + 1
if maxSum1 + sum2 > maxSum12:
maxSum12 = maxSum1 + sum2
maxSum12Idx = (maxSum1Idx, i - k * 2 + 1)
if maxSum12 + sum3 > maxTotal:
maxTotal = maxSum12 + sum3
ans = [*maxSum12Idx, i - k + 1]
sum1 -= nums[i - k * 3 + 1]
sum2 -= nums[i - k * 2 + 1]
sum3 -= nums[i - k + 1]
return ans
推荐阅读
- #|算法理论——快速幂思想(附例题)
- #|Python基础——强制类型转换及字符串
- 矩阵|矩阵分析与应用+张贤达
- 机器学习|吴恩达机器学习(五)神经网络 2/2 —— 反向传播算法(BP-神经网络)
- #|Spring Boot spring.factories 用法及原理
- 算法|基于Matlab的多模态医学图像融合仿真
- #|微信小程序 + 腾讯位置服务获取全国城市列表
- 数据分析|最小二乘和极大似然估计的原理,思想(相同点以及异同?)
- 自动化测试|离职前一定要做好这7件事情,少一件都很麻烦。