bzoj 4385: [POI2015]Wilcze doły 单调队列

题意

给定一个长度为n的序列，你有一次机会选中一段连续的长度不超过d的区间，将里面所有数字全部修改为0。
请找到最长的一段连续区间，使得该区间内所有数字之和不超过p。
1<=d<=n<=2000000，0<=p<=10^16，1<=w[i]<=10^9

分析

首先考虑没有机会时要怎么做，显然可以用双指针乱搞。
那加上了机会呢？显然也是可以双指针乱搞。
显然随着右端点不断递增，左端点是不递减的。考虑如何维护左端点。
显然一个区间[l,r]用掉机会的最小和等于s[r]-s[l-1]-max(c[l+d-1..r])
c[i]表示sum(a[i-d+1..i])
那么用单调队列维护c的最大值即可。

代码

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>using namespace std;typedef long long LL;const int N=2000005;int n,d,head,tail,q2[N];LL q1[N],a[N],p;int read(){    int x=0,f=1;char ch=getchar();    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}    return x*f;}int main(){    n=read();scanf("%lld",&p);d=read();    for (int i=1;i<=n;i++) a[i]=read(),a[i]+=a[i-1];    int ans=d,l=1,r=d,head=1,tail=1;q1[tail]=a[d];q2[tail]=d;    while (r<n)    {        r++;        while (head<=tail&&q1[tail]<=a[r]-a[r-d]) tail--;        q1[++tail]=a[r]-a[r-d];q2[tail]=r;        while (a[r]-a[l-1]-q1[head]>p)        {            if (head<=tail&&q2[head]==l+d-1) head++;            l++;        }        ans=max(ans,r-l+1);    }    printf("%d",ans);    return 0;}