NOIP提高组【JZOJ4811】排队

Description

Data Constraint

Solution

对于询问一，我们只要用堆来查看一下当前没有人的优先级最小的是哪个房间，逐个放进去就好。那对于询问二，我们用LCA计算一下该点到根的路径上的离当前点最远的点，将它往下跳一个格即可。

代码

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=400005;
int first[maxn],last[maxn],next[maxn],fa[maxn],a[maxn],f[maxn][20];
int n,m,i,t,j,k,l,x,y,num,d[maxn],ans,bz[maxn],c[maxn],dfn[maxn],num1,p;
bool bz1[maxn];char ch;
void lian(int x,int y){
    last[++num]=y;next[num]=first[x];first[x]=num;
}
void up(int x){
    while (x>1 && dfn[d[x]]<dfn[d[x/2]]){
        swap(d[x],d[x/2]);
        c[d[x]]=x,c[d[x/2]]=x/2;
        x/=2;
    }
}
void down(int x){
    int y;
    while (x*2<=d[0] && dfn[d[x]]>dfn[d[x*2]] || x*2<d[0] && dfn[d[x]]>dfn[d[x*2+1]]){
        y=x*2;
        if (y<d[0] && dfn[d[y+1]]<dfn[d[y]]) y++;
        swap(d[x],d[y]);c[d[x]]=x,c[d[y]]=y;
        x=y;
    }
}
void dg(int x,int y){
    int t,i,k=0;fa[x]=y;f[x][0]=y;
    for (t=first[x];t;t=next[t]){
        if (last[t]==y) continue;
        a[++a[0]]=last[t];
    }
    sort(a+1,a+a[0]+1);first[x]=0;
    while (a[0]) lian(x,a[a[0]--]);
    for (t=first[x];t;t=next[t]){
        if (last[t]==y) continue;
        k++;
        dg(last[t],x);
    }
    if (!k) d[++d[0]]=x,c[x]=d[0],bz[x]++;
    dfn[x]=++num1;
}
int main(){
//  freopen("data.in","r",stdin);freopen("data.out","w",stdout);
    scanf("%d%d",&n,&m);
    for (i=1;i<n;i++)
        scanf("%d%d",&x,&y),lian(x,y),lian(y,x);
    dg(1,0);l=log(n)/log(2);
    for (j=1;j<=l;j++)
        for (i=1;i<=n;i++)
            f[i][j]=f[f[i][j-1]][j-1];
    num=0;
    for (i=1;i<=m;i++){
        scanf("%d%d",&x,&y);
        if (x==1){
            for (j=1;j<=y;j++){
                t=d[1];bz1[d[1]]=true;
                d[1]=d[d[0]];c[d[d[0]]]=1;
                d[0]--;
                down(1);
                if (!bz[fa[t]] && fa[t]) d[++d[0]]=fa[t],c[fa[t]]=d[0],up(d[0]);
                if (fa[t]) bz[fa[t]]++;
            }
            printf("%d\n",t);   
        }else{
            t=y;ans=0;
            for (j=l;j>=0;j--)
                if (bz1[f[t][j]]) t=f[t][j],ans+=1<<j;
            printf("%d\n",ans);
            bz1[t]=false;p=fa[t];
            if (bz[p]==1){
                c[t]=c[p];
                d[c[t]]=t;
                up(c[t]);c[p]=0;
            }else d[++d[0]]=t,c[t]=d[0],up(d[0]);
            if (p) bz[p]--;

        }
    }
}