题目大意:
给出一个N个数的整数数列:A1,A2,A3……。选择长度不超过k的连续区间,使总和最大。
输入样例:
3 2
1 2 3
输出样例:
5
1<=k<=n<=10^5,Ai<=10^9
解题思路
这题是一题单调队列的模板题,先记录第i个数的前缀和,我们维护一个元素单调递增的队列,每次处理完一个r后把队尾所有不小于sum[r]的元素删掉后把sum[r]放进队尾。那么每次只用把队头中元素在原数组中的下标<r-k的元素删掉,然后选取队头即可得到以r为右端点的最大总和。
辣鸡程序(对C++不熟):
#include<cstdio>
#define MAXN 100005
using namespace std;
long long q[MAXN], sum[MAXN];
int head,tail,i,n,x,k;
long long ans;
int main()
{
scanf("%d%d",&n,&k);
head=1;tail=1;
for (int i=1;i<=n;i++)
{
scanf("%d",&x);
sum[i]=sum[i-1]+x;
while (sum[i]<=sum[q[tail]]&&tail>=head) tail--;
q[++tail]=i;
while (q[head]<i-k) head++;
if ((sum[i]-sum[q[head]])>ans) ans=sum[i]-sum[q[head]];
}
printf("%lld",ans);
return 0;
}
PS 这是一道单调队列模板题