cf102452I. Incoming Asteroids

题目描述

题解

如果说这 $k$ 个的和超过照相机的时间 $s$ ，那至少有一个值 $\ge s/k$ ，其中 $s/k$ 取上整。

所以对每个星星维护一个堆，按照 $s/k$ 从小到大排序，每次取出看是否符合条件即可。

每次取出， $s$ 至多变为 $2/3\times s$ ，所以效率为 $O(m{\log }_{2}m{\log }_{2/3}s)$ 。

代码

#include<bits/stdc++.h>
#define LL long long
using namespace std;
const int N=2e5+5;
int n,m,t,p[4][N],c[N];
LL a[N],b[N];bool d[N];
struct O{int x;LL v;};
bool operator < (O A,O B){return A.v>B.v;}
priority_queue<O>q[N];
int main(){
	scanf("%d%d",&n,&m);
	for (int op,x,y,lst=0;m--;){
		scanf("%d%d%d",&op,&x,&y);
		if (op&1){
			x^=lst;p[0][++t]=y;b[t]=x;
			for (int j=1,u;j<=y;j++){
				scanf("%d",&u);u^=lst;
				p[j][t]=u;b[t]+=a[u];
				q[u].push((O){t,a[u]+(x+y-1)/y});
			}
			continue;
		}
		x^=lst;y^=lst;a[x]+=y;lst=0;
		for (;!q[x].empty();){
			O u=q[x].top();q[x].pop();
			if (u.v>a[x]){q[x].push(u);break;}
			if (d[u.x]) continue;LL v=0;
			for (int i=1;i<=p[0][u.x];i++) v+=a[p[i][u.x]];
			if (v>=b[u.x]) c[++lst]=u.x,d[u.x]=1;
			else for (int i=1,k,j=p[0][u.x];i<=j;i++)
				k=p[i][u.x],
				q[k].push((O){u.x,(b[u.x]-v+j-1)/j+a[k]});
		}
		printf("%d",lst);
		sort(c+1,c+lst+1);
		for (int i=1;i<=lst;i++) printf(" %d",c[i]);
		puts("");
	}
	return 0;
}