數數

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