hdu 2430 Beans 单调队列

来源：互联网发布：linux 卸载python 编辑：程序博客网时间：2024/05/14 04:45
#include<iostream>#include<cstdio>#include<vector>#include<cstring>#include<string>#include<algorithm>#include<queue>#include<cmath>using namespace std;//题意：从num[1]~num[n]中取一段连续的序列，使得（num[i]+...+num[j]）%p<=k且该序列的和最大化//令ma=a%mod,mb=b%mod,如果ma<mb,那么(a-b)%mod的值会大于ma//所以要想(a-b)%mod小于ma，那么ma要大于mb//这里使用单调队列就是要维护一个点前缀和sum[i]%p单调递增//在使得sum[i]%p是单调的前提下维护位置是递增的，这样才会使得队尾元素sum[tail]与队首元素sum[head]的差值最大，即区间和最大const int maxn=1000005;int num[maxn];long long sum[maxn];struct node{int pos,mod;node(int p,int m):pos(p),mod(m){}node(){}}a[maxn],que[maxn];bool cmp(const node& a,const node& b) {if(a.mod==b.mod) return a.pos<b.pos;return a.mod<b.mod;}int main(){int Case;scanf("%d",&Case);for(int kase=1;kase<=Case;kase++){int n,p,k;scanf("%d%d%d",&n,&p,&k);sum[0]=0;for(int i=1;i<=n;i++){scanf("%d",&num[i]);sum[i]=sum[i-1]+num[i];a[i].mod=sum[i]%p;a[i].pos=i;}sort(a+1,a+1+n,cmp);long long ans=-1;int head=0,tail=0;for(int i=1;i<=n;i++){if(tail>head&&que[tail-1].pos>a[i].pos)tail--;//这样为了保证队列中的点的pos是单调的，只有这样，队尾减去队首的差值才会最大que[tail++]=a[i];while(tail>head&&a[i].mod-que[head].mod>k) head++;if(a[i].mod<=k) ans=max(ans,sum[a[i].pos]);else if(head<tail-1){ans=max(ans,sum[a[i].pos]-sum[que[head].pos]);//cout<<a[i].pos<<endl;}}if(ans!=-1) ans=ans/p;printf("Case %d: %I64d\n",kase,ans);}}
0 0