SSL 2521 数数

Description

给出一个长度为N的整数序列，选择长度不超过K的段，使总和最大

Simple Input

3 2
1 2 3

Simple Output

5

Hint

1<=K<=N<=10^5
abs(a[i])<=10^9

分析

方法一：枚举左端点l和右端点r，然后扫一遍计算总和并选取最大值。时间复杂度O(n^3)
方法二：预处理sum[i]表示前i个数的和，枚举左端点l和右端点r，然后O(1)计算总和并选取最大值。时间复杂度O(n^2)
方法三：仍然是预处理sum[i]表示前i个数的和，只枚举右端点r，问题就转换成了在sum[r-k..r-1]中找一个最小值。那么我们维护一个元素单调递增的队列，每次处理完一个r后把队尾所有不小于sum[r]的元素删掉后把sum[r]放进队尾。那么每次只用把队头中元素在原数组中的下标

代码

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

#define ll long long
#define N 500005
#define INF 0x7fffffff

ll read()
{
    ll x=0,k=1;
    char ch;
    ch=getchar();
    while(ch<'0' || ch>'9')
    {
        if(ch=='-')
            k=-1;
        ch=getchar();
    }
    while(ch>='0' && ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*k;
}

ll a[N],f[N];
ll sum[N];
ll q[N];

int main()
{
    ll n,k;
    n=read();
    k=read();
    int head=0,tail=0;
    ll ans=-INF;
    for(int i=1;i<=n;i++)
    {
        a[i]=read();
        sum[i]=a[i]+sum[i-1];
        while(head<=tail && i-k-1>=q[head])
            head++;
        f[i]=sum[i]-sum[q[head]];
        while(head<=tail && sum[q[tail]]>=sum[i])
            tail--;
        q[++tail]=i;
        ans=std::max(ans,f[i]);
    }
    std::cout<<ans<<std::endl;
}