BZOJ4385[POI2015] Wilcze doły

BZOJ4385[POI2015] Wilcze doły

Description

給定一個長度爲n的序列，你有一次機會選中一段連續的長度不超過d的區間，將裏面所有數字全部修改爲0。

請找到最長的一段連續區間，使得該區間內所有數字之和不超過p。

Input

第一行包含三個整數n,p,d(1<=d<=n<=2000000，0<=p<=10^16)。

第二行包含n個正整數，依次表示序列中每個數wi。

Output

包含一行一個正整數，即修改後能找到的最長的符合條件的區間的長度。

Sample Input

9 7 2

3 4 1 9 4 1 7 1 3

Sample Output

5

Hint

將第4個和第5個數修改爲0，然後可以選出區間[2,6]，總和爲4+1+0+0+1=6。

Solution：

首先顯然是尺取，然後對於這個區間中，我們尋找一個區間把它變成零，也就是要在這個大區間找一個長度爲d 的小區間使得區間和最大，用單調隊列維護一下就好了，複雜度O(n) 。

#include<stdio.h>
#include<iostream>
#define ll long long
#define M 2000005
#define mp(a,b) (Node){a,b}
using namespace std;
int A[M];
ll sum[M],K[M];
struct Node{ll x;int id;}Q[M];
int l=1,r=0;
inline void Rd(int &res){
    char c;res=0;
    while(c=getchar(),!isdigit(c));
    do{
        res=(res<<3)+(res<<1)+(c^48);
    }while(c=getchar(),isdigit(c));
}
int main(){
    int n,d,ans=0;ll q;
    Rd(n);cin>>q;Rd(d);
    for(int i=1;i<=n;i++){
        Rd(A[i]);
        sum[i]=sum[i-1]+A[i];
    }
    for(int i=1;i<=n-d+1;i++)
        K[i]=sum[i+d-1]-sum[i-1];
    Q[++r]=mp(K[1],1);
    int L=1,R=d;
    while(R<=n){
        while(l<=r){
            if(Q[l].id<L){l++;continue;}
            if(sum[R]-sum[L-1]-Q[l].x>q){L++;continue;}
            break;
        }
        if(R-L+1>ans)ans=R-L+1;
        R++;
        while(l<=r&&K[R-d+1]>Q[r].x)r--;
        Q[++r]=mp(K[R-d+1],R-d+1);
    }
    printf("%d\n",ans);
    return 0;
}