ps:参考例子,自己打了一遍,没有算法创新
给你一个整数数组 nums 和一个整数 k ,找出三个长度为 k 、互不重叠、且 3 * k 项的和最大的子数组,并返回这三个子数组。
以下标的数组形式返回结果,数组中的每一项分别指示每个子数组的起始位置(下标从 0 开始)。如果有多个结果,返回字典序最小的一个。
class Solution {
public:
vector<int> maxSumOfThreeSubarrays(vector<int>& nums, int k)
{
vector<int> ans;
int sum1=0, MaxSum1=0, index1=0;
int sum2=0, MaxSum2=0, index2_1=0,index2_2;
int sum3 = 0, MaxSum3=0, index3=0;
for(int i=0;i+2*k<nums.size();i++)
{
//每次循环加上一个新的数据
sum1+=nums[i];
sum2+=nums[i+k];
sum3+=nums[i+2*k];
if(i>=k-1)//判断是否每个sum都包含k个数
{
if(sum1>MaxSum1)
{
MaxSum1 = sum1;
index1 = i-k+1;
}
if(MaxSum1+sum2>MaxSum2)
{
MaxSum2=MaxSum1+sum2;
//保证序号不重复
//我刚开始没有存第一个数的下标,发现会造成重复,因为第一个数在不断更新,下标也在更新,但后面两个数有时没有更新,导致错误
index2_1=index1;
index2_2=i+1;
}
if(MaxSum2+sum3>MaxSum3)
{
MaxSum3=MaxSum2+sum3;
index3=i+k+1;
ans = {index2_1,index2_2,index3};
}
//每次循环减去最前面的数据,每个和由保持k个数组成
sum1-=nums[i-k+1];
sum2-=nums[i+1];
sum3-=nums[i+k+1];
}
}
return ans;
}
};
执行结果:通过
执行用时:16 ms, 在所有 C++ 提交中击败了96.57%的用户
内存消耗:19.6 MB, 在所有 C++ 提交中击败了99.43%的用户
通过测试用例:43 / 43
小声bb:题目越来越看不懂了,绝了