数数

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);}