luogu2827 [NOIp2016]蚯蚓 (模拟)
可以直观地想到用优先队列来做，但数据范围是O(n)的
然后我们发现，因为我们每次挑出来的蚯蚓是单调的，所以把每个切成两段后，那两段也是对应单调的
也就是说，算上最一开始的蚯蚓，我们一共维护三个队列，三个分别单调，每次取出来最大的那个队头就行了
1 #include<bits/stdc++.h> 2 #define pa pair<int,int> 3 #define CLR(a,x) memset(a,x,sizeof(a)) 4 using namespace std; 5 typedef long long ll; 6 const int maxn=1e5+10,maxm=7e6+10; 7 const ll inf=1e15; 8 9 inline ll rd(){ 10 ll x=0;char c=getchar();int neg=1; 11 while(c<'0'||c>'9'){if(c=='-') neg=-1;c=getchar();} 12 while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar(); 13 return x*neg; 14 } 15 16 int T; 17 int num[maxn]; 18 ll N,M,q; 19 double p; 20 queue<ll> q1,q2,q3; 21 22 int main(){ 23 //freopen(".in","r",stdin); 24 int i,j,k; 25 N=rd(),M=rd();q=rd(); 26 p=(double)rd()/rd();T=rd(); 27 for(i=1;i<=N;i++) num[i]=rd(); 28 sort(num+1,num+N+1); 29 for(i=N;i;i--) q1.push(num[i]); 30 for(i=1;i<=M;i++){ 31 ll a=(q1.empty()?-inf:q1.front()),b=(q2.empty()?-inf:q2.front()),c=(q3.empty()?-inf:q3.front()); 32 ll mm=max(a,max(b,c)); 33 34 if(a==mm) q1.pop(); 35 else if(b==mm) q2.pop(); 36 else if(c==mm) q3.pop(); 37 mm+=q*(i-1); 38 ll x=(ll)(mm*p),y=mm-x; 39 q2.push(x-q*i),q3.push(y-q*i); 40 if(i%T==0) printf("%lld ",mm); 41 }printf("\n"); 42 43 for(i=1;(!q1.empty())||(!q2.empty())||(!q3.empty());i++){ 44 ll a=(q1.empty()?-inf:q1.front()),b=(q2.empty()?-inf:q2.front()),c=(q3.empty()?-inf:q3.front()); 45 ll mm=max(a,max(b,c)); 46 // printf("%lld %lld %lld\n",a,b,c); 47 if(mm==-inf) break; 48 if(q1.front()==mm) q1.pop(); 49 else if(q2.front()==mm) q2.pop(); 50 else if(q3.front()==mm) q3.pop(); 51 mm+=q*M; 52 if(i%T==0) printf("%lld ",mm); 53 } 54 55 return 0; 56 }
