数数 ssl 2521 单调队列

题目大意

给出一个长度为n(1<=n<=100000)的整数序列，选择长度不超过k(1<=k<=n)的段，使得总和最大。

分析

和烽火传递一样，优先队列水过~~

code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<string>
#include<algorithm>

using namespace std;

long long num[200000];
long long f[200000];
long long a[200000];
int queue[200000];
int head,tail;
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    num[1]=a[1];
    for (int i=2;i<=n;i++)
        num[i]=num[i-1]+a[i];
    head=1; tail=1;
    queue[head]=0;
    for (int i=1;i<=n;i++)
    {
        while ((queue[head]<i-m)&&(tail>=head))
            head++;
        while ((num[queue[tail]]>num[i])&&(tail>=head))
            tail--;
        tail++;
        queue[tail]=i;
        f[i]=num[i]-num[queue[head]];
    }
    long long ans=0;
    for (int i=1;i<=n;i++)
        ans=max(f[i],ans);
    printf("%lld",ans);
}