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;
}