给你一个整数数组 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.滑动窗口法。
单个子数组的最大和。从[0,k−1]开始,不断地向右滑动窗口,直至窗口右端点到达数组末尾
时停止。
class Solution {
public int[] maxSumOfOneSubarray(int[] nums, int k) {
int[] ans = new int[1];//存储窗口起始下标
int sum1 = 0, maxSum1 = 0;
for (int i = 0; i < nums.length; ++i) {
sum1 += nums[i];
if (i >= k - 1) {//说明当前窗口内已经有了k个数
if (sum1 > maxSum1) {//如果当前窗口内的和大于之前存储的最大和
maxSum1 = sum1;//更新最大和
ans[0] = i - k + 1;//存储当前窗口的第一个元素
}
sum1 -= nums[i - k + 1];//继续向后移动,保持窗口内元素个数为k,减去当前窗口的第一个元素
}
}
return ans;//返回下标
}
}
两个无重叠子数组的最大和。维护第一个子数组的最大值,然后向后遍历第二个子数组,
找到最大值。
class Solution {
public int[] maxSumOfTwoSubarrays(int[] nums, int k) {
int[] ans = new int[2];//存储窗口起始下标
int sum1 = 0, maxSum1 = 0, maxSum1Idx = 0;
int sum2 = 0, maxSum12 = 0;
for (int i = k; i < nums.length; ++i) {//初始值i=k,第一个窗口从0开始,即i-k,第二个窗口从k开始,即i
sum1 += nums[i - k];
sum2 += nums[i];
if (i >= k * 2 - 1) {//当两个窗口中元素都满了
if (sum1 > maxSum1) {
maxSum1 = sum1;//更新第一个窗口的最大值
maxSum1Idx = i - k * 2 + 1;//更新第一个窗口最大和的起始下标
}
if (maxSum1 + sum2 > maxSum12) {//更新两个窗口总和的最大值
maxSum12 = maxSum1 + sum2;
ans[0] = maxSum1Idx;
ans[1] = i - k + 1;
}
//继续向后遍历,分别减去第一个窗口、第二个窗口的起始元素
sum1 -= nums[i - k * 2 + 1];
sum2 -= nums[i - k + 1];
}
}
return ans;
}
}
三个无重叠子数组的最大和。同时向右滑动这三个窗口,按照前言二的方法并维护
maxSum12及其对应位置。每次滑动时,计算当前maxSum12与sum 3之和。统计这一过程中
的maxSum12+sum3的最大值及其对应位置。
class Solution {
public int[] maxSumOfThreeSubarrays(int[] nums, int k) {
int[] ans = new int[3];
int sum1 = 0, maxSum1 = 0, maxSum1Idx = 0;
int sum2 = 0, maxSum12 = 0, maxSum12Idx1 = 0, maxSum12Idx2 = 0;
int sum3 = 0, maxTotal = 0;
for (int i = k * 2; i < nums.length; ++i) {
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;
maxSum12Idx1 = maxSum1Idx;
maxSum12Idx2 = i - k * 2 + 1;
}
if (maxSum12 + sum3 > maxTotal) {
maxTotal = maxSum12 + sum3;
ans[0] = maxSum12Idx1;
ans[1] = maxSum12Idx2;
ans[2] = i - k + 1;
}
sum1 -= nums[i - k * 3 + 1];
sum2 -= nums[i - k * 2 + 1];
sum3 -= nums[i - k + 1];
}
}
return ans;
}
}
还有另一种动态规划方法,但是自己暂时没看懂。
参考:Java实现 LeetCode 689 三个无重叠子数组的最大和(换方向筛选)_南 墙-CSDN博客
题源:力扣