排队

题目描述

做法

我们先处理出fix表示在全空情况下不断进人，i这个房间被第几个人最终停留。
用一个堆保留空房间当前fix的最小值，每次进一个人就是选择堆中最小值的房间。
拿走人的话观察一下就是从这个房间往上有人房间的数量-1，这个可以倍增来求，因为这个房间到根路径上有人和没人一定是分离开的。
分析一下势能这样做是n log n的。

#include<cstdio>
#include<algorithm>
#include<set>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int maxn=100000+10;
int f[maxn][25],fix[maxn],co[maxn],d[maxn],b[maxn*2],sta[100];
int h[maxn],go[maxn*2],next[maxn*2];
struct dong{
    int id,fix;
    friend bool operator <(dong a,dong b){
        return a.fix<b.fix;
    }
};
set<dong> s;
dong zlt;
int i,j,k,l,t,n,m,tot,top;
int read(){
    int x=0;
    char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x;
}
void add(int x,int y){
    go[++tot]=y;
    next[tot]=h[x];
    h[x]=tot;
}
void dfs(int x,int y){
    f[x][0]=y;
    d[x]=d[y]+1;
    int i,t=h[x],l=top+1,r=top;
    while (t){
        if (go[t]!=y) b[++r]=go[t];
        t=next[t];
    }
    sort(b+l,b+r+1);
    top=r;
    fo(i,l,r) dfs(b[i],x);
    top=l-1;
    fix[x]=++tot;
}
int getfather(int x){
    int j=floor(log(n)/log(2));
    while (j>=0){
        if (co[f[x][j]]==1) x=f[x][j];
        j--;
    }
    return x;
}
void write(int x){
    if (!x){
        putchar('0');
        putchar('\n');
        return;
    }
    top=0;
    while (x){
        sta[++top]=x%10;
        x/=10;
    }
    while (top){
        putchar('0'+sta[top]);
        top--;
    }
    putchar('\n');
}
int main(){
    n=read();m=read();
    fo(i,1,n-1){
        j=read();k=read();
        add(j,k);add(k,j);
    }
    top=tot=0;
    dfs(1,0);
    fo(j,1,floor(log(n)/log(2)))
        fo(i,1,n)
            f[i][j]=f[f[i][j-1]][j-1];
    fo(i,1,n){
        zlt.id=i;
        zlt.fix=fix[i];
        s.insert(zlt);
    }
    while (m--){
        t=read();
        if (t==1){
            k=read();
            while (k--){
                zlt=*s.begin();
                co[zlt.id]=1;
                if (!k) write(zlt.id);
                s.erase(s.find(zlt));
            }
        }
        else{
            k=read();
            if (!co[k]){
                write(0);
                continue;
            }
            j=getfather(k);
            write(d[k]-d[j]);
            co[j]=0;
            zlt.id=j;
            zlt.fix=fix[j];
            s.insert(zlt);
        }
    }
}