无重叠子数组的最大和【滑动窗口】

题目是力扣P689三个无重叠子数组的最大和
在这里插入图片描述
首先我们考虑解决长度为k的单个子数组的最大和问题:
假设 s u m 1 sum_1 sum1是大小为k的窗口的元素和,当窗口从[i-k+1,i]向右滑动一个元素后, s u m 1 sum_1 sum1增加了nums[i+1],减少了nums[i-k+1]。我们从0开始,增大窗口直至窗口大小为k,然后不断向右滑动窗口,直到窗口的右端点到达数组末尾时结束,统计这以过程中的 s u m 1 sum_1 sum1最大值(记作maxsum1)及其位置就行。

vector<int> maxSumOfOneSubarray(vector<int> &nums, int k) {
	int n=nums.size();
	int ans;
	int sum1=0,maxsum1=0;
	for(int i=0;i<n;i++){
		sum1+=nums[i];
		if(i>=k-1){ //窗口大小等于k
			if(sum1>maxsum1){
				maxsum1=sum1;
				ans=i-k+1 //i记录的是窗口右端的位置
			}
			sum1-=nums[i-k+1];
		}
	}
	return ans; //返回最大的位置
}

我们再考虑两个无重叠的子数组的最大和,如果我们使用两个大小为k的滑动窗口,假设 s u m 1 sum_1 sum1为第一个滑动窗口的元素和,该窗口从i=k-1开始滑动, s u m 2 sum_2 sum2为第二个滑动窗口的元素和,该窗口从i=2*k-1开始滑动。我们同时向右滑动这两个窗口,并维护 s u m 1 sum_1 sum1的最大值 m a s S u m 1 masSum_1 masSum1及其对应位置。每次滑动时,计算当前 m a s S u m 1 masSum_1 masSum1 s u m 2 sum_2 sum2之和。统计这一过程中的 m a x S u m 1 maxSum_1 maxSum1+ s u m 2 sum_2 sum2的最大值(记作 m a x S u m 12 maxSum_{12} maxSum12及其对应位置

vector<int> maxSumOfTwoSubarrays(vector<int> &nums, int k) {
	int n=nums.size();
	vector<int>ans;
	int sum1=0,maxSum1=0,maxSum1Idx=0;
	int sum2=0,maxSum2=0;
	for(int i=k;i<n;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={maxSumIdx,i-k+1};
			}
			sum1-=nums[i-k*2+1];
			sum2-=nums[i-k+1];
		}
	}
	return ans;
}

会到本题,我们需要使用3个大小为k的滑动窗口,同理,假设 s u m 1 sum_1 sum1为第一个滑动窗口的元素和,该窗口从i=k-1开始滑动, s u m 2 sum_2 sum2为第二个滑动窗口的元素和,该窗口从i=2k-1开始滑动, s u m 3 sum_3 sum3为第三个滑动窗口的元素和,该窗口从i=3k-1开始滑动。
我们同时向右滑动这三个窗口,每次滑动时,按照前面的方法维护 m a x S u m 12 maxSum_{12} maxSum12及其对应位置,计算当前 m a x S u m 12 maxSum_{12} maxSum12+ s u m 3 sum_3 sum3,统计这一过程中 m a x S u m 12 maxSum_{12} maxSum12+ s u m 3 sum_3 sum3的最大值及其位置即可。

vector<int> maxSumOfThreeSubarrays(vector<int> &nums, int k) {
	vector<int>ans;
	int n=nums.size();
	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<n;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={maxSum12Idx1,maxSum12Idx2,i-k+1};
			}
			sum1-=nums[i-k*3+1];
			sum2-=nums[i-k*2+1];
			sum3-=nums[i-k+1];
		}
	}
	return ans;
}

好麻烦的题,但是想懂了就应该能写出来,不愧是困难题

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值