Description
给出一个长度为N的整数序列,选择长度不超过K的段,使总和最大
Simple Input
3 2
1 2 3
Simple Output
5
Hint
1<=K<=N<=10^5
abs(a[i])<=10^9
解题思路
先预处理前缀和sum[i],每段的和等于sum[i]-sum[x],就是用一个队列来做。每次把队头取出来,队头若小于i-k那么就出队,每次队尾若位置小于sum[i]但位置比i前的就删掉。
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include<cmath>
const int N=100001;
long long sum[N],p[N],a[N];
using namespace std;
int main()
{
long long n,tot,k,ans;
scanf("%lld%lld",&n,&k);
for (int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
sum[i]=sum[i-1]+a[i];
}
int head,tail;
head=1;
tail=1;
ans=a[1];
for (int i=1;i<=n;i++)
{
while (p[head]<=i-k-1 && head<=tail)
head++;
tot=sum[i]-sum[p[head]];
ans=max(tot,ans);
while (sum[p[tail]]>=sum[i] && (head<=tail))
tail--;
p[++tail]=i;
}
printf("%lld\n" ,ans);
}