hdu 2430 Beans 單調佇列
阿新 • • 發佈:2019-01-29
#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); } }