给你一个整数数组 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 * 104
1 <= nums[i] < 2^16
1 <= k <= floor(nums.length / 3)
题解:先考虑一下如果是求一个长度为K的连续子序列最大和,比较简单,直接滑动窗口,l为滑动窗口的左边,r为滑动窗口的右边,求一下就好了。如果是求两个,需要两个滑动窗口,求三个就是三个滑动窗口就好了。
import java.util.*;
import java.util.Map.Entry;
class Solution {
public int[] maxSumOfThreeSubarrays(int[] nums, int k) {
int len=nums.length;
int ans[]=new int[3];
int fsum=0,findex=0,fmaxSum=0;
int ssum=0,fsindex=0,sindex=0,smaxSum=0;
int tsum=0,maxTotal=0;
for(int i=2*k;i<len;i++)
{
fsum+=nums[i-2*k];
ssum+=nums[i-k];
tsum+=nums[i];
if(i>=3*k-1)
{
if(fsum>fmaxSum)
{
fmaxSum=fsum;
findex=i-3*k+1;
}
if(fmaxSum+ssum>smaxSum)
{
smaxSum=fmaxSum+ssum;
fsindex=findex;
sindex=i-2*k+1;
}
if(smaxSum+tsum>maxTotal)
{
maxTotal=smaxSum+tsum;
ans[0]=fsindex;
ans[1]=sindex;
ans[2]=i-k+1;
}
fsum-=nums[i-3*k+1];
ssum-=nums[i-2*k+1];
tsum-=nums[i-k+1];
}
}
return ans;
}
}