题意:
给出一个序列,你能将一个长度不超过d的连续子序列全部变为0,要求和不超过p的最长连续子序列。序列大小≤2000000。
题解:
用两个指针,每次右指针右移时就将新加入元素所能消掉的区间加入单调队列,如果当前区间和减单调队列中最大的元素不超过p,则累计答案,否则左指针左移并在单调队列中删掉这个元素(如果存在的话)。反思:本弱wa了4发,对拍查不出错误,后来和标程一点一点对比,才发现问题。原因是ans没有初始化为d(见我的烂代码),这样的话如果遇到这个数据就会挂:
2 0 1
1 2
结果会输出0。不过根本原因是我程序写得不好,容易出bug。
代码:
我的烂代码:
1 #include <cstdio> 2 #include <cstring> 3 #include <algorithm> 4 #define inc(i,j,k) for(int i=j;i<=k;i++) 5 #define maxn 2000010 6 #define ll long long 7 using namespace std; 8 9 inline ll read(){ 10 char ch=getchar(); ll f=1,x=0; 11 while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();} 12 while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar(); 13 return f*x; 14 } 15 ll a[maxn],sum[maxn],q1[maxn],now,p; int q2[maxn],l,r,ql,qr,ans,n,d; 16 int main(){ 17 n=read(); p=read(); d=read(); 18 if(d>=n){printf("%d",n); return 0;} 19 inc(i,1,n)a[i]=read(),sum[i]=sum[i-1]+a[i]; 20 l=1; r=d; ql=qr=1; q1[1]=now=sum[r]-sum[l-1]; q2[1]=1; ans=d; 21 while(1){ 22 while(r<n&&(r-l+1<d||now<=p+q1[ql])){ 23 r++; now+=a[r]; 24 while(ql<=qr&&sum[r]-sum[r-d]>=q1[qr])qr--; 25 q1[++qr]=sum[r]-sum[r-d]; q2[qr]=r-d+1; 26 if(now<=p+q1[ql])ans=max(ans,r-l+1); 27 } 28 if(r==n)break; now-=a[l]; if(ql<=qr&&q2[ql]<=l)ql++; l++; 29 } 30 printf("%d",ans); return 0; 31 }
照标程写的代码:
1 #include <cstdio> 2 #include <cstring> 3 #include <algorithm> 4 #define inc(i,j,k) for(int i=j;i<=k;i++) 5 #define maxn 2000010 6 #define ll long long 7 using namespace std; 8 9 inline ll read(){ 10 char ch=getchar(); ll f=1,x=0; 11 while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();} 12 while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar(); 13 return f*x; 14 } 15 ll a[maxn],sum[maxn],q1[maxn],now,p; int q2[maxn],l,r,ql,qr,ans,n,d; 16 int main(){ 17 n=read(); p=read(); d=read(); 18 if(d>=n){printf("%d",n); return 0;} 19 inc(i,1,n)a[i]=read(),sum[i]=sum[i-1]+a[i]; 20 l=1; r=d; ql=qr=1; q1[1]=now=sum[r]-sum[l-1]; q2[1]=1; 21 while(r<n){ 22 r++; now+=a[r]; while(ql<=qr&&sum[r]-sum[r-d]>=q1[qr])qr--; 23 q1[++qr]=sum[r]-sum[r-d]; q2[qr]=r-d+1; 24 while(now>p+q1[ql]){now-=a[l]; if(q2[ql]<=l)ql++; l++;} 25 ans=max(ans,r-l+1); 26 } 27 printf("%d",ans); return 0; 28 }
20160823