[后序遍历][堆][倍增] Jzoj P4811

 

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

 

 

题解

• 首先，我们很容易就可以发现，每个人的走法就是后序遍历的走法
• 询问删掉一个人之后又多少个人变化，也就是该点到根路径上人的数量
• 那么我们就可以用堆维护房间，然后用倍增往上跳找人数

代码

#include <cstdio>
#include <iostream>
#include <cstring>
#include <vector>
#include <cmath>
#include <algorithm>
using namespace std;
const int N=100010;
int n,t,tot,cnt,ans,dfx[N],d[N],p[N],heap[N*4],deep[N],f[N][20];
bool bz[N];
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;
}
void insert(int x)
{
    heap[++tot]=x;
    for (int i=tot;i>1&&p[heap[i]]<p[heap[i/2]];i/=2) swap(heap[i],heap[i/2]);
}
int get()
{
    int x=heap[1],i=1;
    heap[1]=heap[tot],heap[tot--]=0;
    while (1)
    {
        int l=i*2,r=i*2+1,m;
        if (!(p[heap[i]]>p[heap[l]]&&l<=tot||p[heap[i]]>p[heap[r]]&&r<=tot)) break;
        if (r>tot||p[heap[l]]<p[heap[r]]) m=l; else m=r;
        swap(heap[i],heap[m]),i=m;
    }
    return x;
}
int jump(int x)
{
    for (int i=log2(deep[x]);i>=0;i--) if (bz[f[x][i]]) x=f[x][i];
    return x;
}
int main()
{
    scanf("%d%d",&n,&t);
    for (int i=1,x,y;i<n;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 j=1;j<=log2(n);j++) for (int i=1;i<=n;i++) f[i][j]=f[f[i][j-1]][j-1];
    sort(d+1,d+n+1,cmp);
    for (int i=1;i<=n;i++) p[d[i]]=i;
    for (int i=1;i<=n;i++) insert(i);
    for (int i=1,x,y,op;i<=t;i++)
    {
        scanf("%d%d",&op,&x);
        if (op==1) 
        {    
            for (;x;x--) bz[ans=get()]=1;
            printf("%d\n",ans);
        }
        else printf("%d\n",deep[x]-deep[y=jump(x)]),insert(y),bz[y]=0;
    }
}