题解
如果说这 $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; }