JZOJ 4811. 排队

题目

Description

 

Input

Output

 

Sample Input

5 4
1 2
1 3
3 4
3 5
1 4
2 4
1 2
2 5

Sample Output

3
1
1
2

 

Data Constraint

 

Hint

 

 

分析

 

 

• 这道题一看应该是个模拟题

• 但是我们需要优化

• 首先我们要跑一个dfs，讲树的优先级跑出来

• 然后我们开一个优先队列

• 讲优先级的下标压进队列

• 然后我们用倍增优化我们的向上找第一个被标记的点

• 结果就是直接的深度

 

代码

#include<iostream>
#include<vector>
#include<algorithm>
#include <queue>
#include<cstdio>
using namespace std;
const int N=100010;
int n,t,tot,cnt,ans,dfx[N],d[N],p[N],deep[N],f[N][20];
bool bz[N];
priority_queue<int,vector<int>,greater<int> > q;
vector<int> a[N];
bool cmp(int x,int y) { return dfx[x]<dfx[y]; }
void dfs(int x,int fa)
{
    f[x][0]=fa,deep[x]=deep[fa]+1;
    for (int i=0;i<a[x].size();i++) if (a[x][i]!=fa) dfs(a[x][i],x); 
    dfx[x]=++cnt;
}
int main ()
{
    cin>>n>>t;
    for (int i=1,x,y;i<=n-1;i++)
    {
        scanf("%d %d",&x,&y); 
        a[x].push_back(y);
        a[y].push_back(x);
    }
    for (int i=1;i<=n;i++) d[i]=i,sort(a[i].begin(),a[i].end());
    dfs(1,0);
    for (int i=1;i<=18;i++)
        for (int j=1;j<=n;j++)
          f[j][i]=f[f[j][i-1]][i-1];
    sort(d+1,d+1+n,cmp);
    for (int i=1;i<=n;i++) p[d[i]]=i;
    for (int i=1;i<=n;i++) q.push(i);
    for (int i=1,x,y,op;i<=t;i++)
    {
        scanf("%d %d",&op,&x);
        if(op==1)
        {
            for (;x;x--) bz[ans=d[q.top()]]=1,q.pop();
            printf("%d\n",ans);
        }
        else
        {
            int a=x;
            for (int i=18;i>=0;i--)
              if (bz[f[a][i]]) a=f[a][i];
            bz[a]=0,q.push(p[a]);
            printf("%d\n",deep[x]-deep[a]);
        }
     } 
}