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