神奇啊最大子序和

最大子序和 sum

——神奇的魔法之单调队列

这道题的神奇之处(个人认为)在于其解题过程中可以用两重循环达到O(n)的效果

Description

输入一个长度为n的整数序列,从中找出一段不超过m的连续子序列,使得整个序列的和最大。 例如 1,-3,5,1,-2,3 当m=4时,S=5+1-2+3=7 当m=2或m=3时,S=5+1=6。

Input

第一行两个数n,m(1<=n,m<=300000) 第二行有n个数,要求在n个数找到最大子序和。

Output

一个数,即最大子序和S(S值不超过long long int)。

不过最后它有多么简单的解法,我第一个想到的是枚举
在这里插入图片描述

三重循环,
先记录前缀和什么的
一二重控制其范围,从挨个递增的起点开始不断记录长度从1到m大小(第三重)的子序和,此过程中对比比出最大值
求子序列和就是对应前缀和相减好啦

这个代码出来的很快,快到我怀疑其正确性(我是有多菜…)

不死心地搞了很多组数据发现真的没问题
在这里插入图片描述

又估计了一下数据觉得不会怎么超时,起码能过一两组小数据吧(为自己的志向跪了OTZ)

于是兴致勃勃地把此代码丢到一边看下一道题去了

意外之喜是评测过了一半数据(剩下的统一超时)
于是就很开心了(^-^)V,人生要有追求,要有可以达到的追求(* ^▽ ^* ),知足常乐吖吖吖
只错了超时说明代码没问题

不过兴奋过后还是唾弃了一下自己ε=(´ο`*))),要是这种枚举都错就已经废了好嘛,代码没问题很正常好嘛。

然后等老师讲
在这里插入图片描述
嗯…果不其然有我没想到的神奇优化

忽然想起做题时有同学趴在某大佬电脑前表示
那个双重循环能过该同学吃屎
然后大佬就过了o(* ̄︶ ̄ *)o

原来如此

首先制作前缀和

	for(int i = 1; i <= n; i++)
	{
		scanf("%d",&a[i]);
		sum[i] = sum[i - 1] + a[i];
	}

求一个子序列就是前缀和相减
由此可得
对于某一个 i,我们要找到一个j(i-j<=m),使得sum[i]-sum[j]最大
那么假设如果有 j1所代表位置在j2之前, 而且sum[j1]>sum[j2],那么j1可以直接抛弃,
(当sum[j1]>sum[j2]时,sum[i]-sum[j1]肯定小于sum[i]-sum[j2])
所以所有j必须是单调递增的,可以用一个单调队列来维护这个 j 的序列

int head = 1, tail = 1;
	q[head] = 0;
	for(int i = 1; i <= n; i++)
	{
		while(head <= tail && q[head] < i-m) 
		{
			head++;
		}
		ans = max(ans, sum[i] - sum[q[head]]);
		while(head <= tail && sum[q[tail]] >= sum[i]) 
		{
			tail--;//残忍地抛弃那个小j吧
		}
		q[+	+tail] = i;
	}

把不需要的j去掉,j的序列里原本到最后会装入从1到n(这个n是最后一次的最后一步时加入的,不会参与运算),经过单调队列的维护会大大加快速度,在i与j之间求得最大子序和,更新最大值,然后输出就好咯
在这里插入图片描述

看上去是两重循环但其实j的队列加加减减是单调增长,所以实际时间复杂度只有O(n)简直不要太爽(^-^)V

以下是完整代码+注释

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int a[300010], q[300010], n, m;
long long sum[300010], ans;

int main()
{
	cin >> n >> m;
	for(int i = 1; i <= n; i++)
	{
		scanf("%d",&a[i]);
		sum[i] = sum[i - 1] + a[i];
	}
	//sum[i]-min{sum[j]|i-m~i-1};
	int head = 1, tail = 1;
	q[head] = 0; //save j,即保存数组下标 
	for(int i = 1; i <= n; i++)
	{
		while(head <= tail && q[head] < i-m) 
		{
			head++;
		}
		ans = max(ans, sum[i] - sum[q[head]]);
		while(head <= tail && sum[q[tail]] >= sum[i]) 
		{
			tail--;//残忍地抛弃那个小j吧
		}
		q[+	+tail] = i;
	}
	cout << ans;
	fclose(stdin);
	fclose(stdout);
	
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值