[CodeForces - 208E] Blood Cousins（k代兄弟问题）

题目链接
加强版例题（还没写）

题意：

给定一棵$n$个结点的树。给出$m$次询问，每次询问给出$x$和$k$，问$x$有多少个$k$代兄弟。（两个点互为$k$代兄弟的定义是他们的$k$级祖先相同）
$1\le n,m\le 10^5$

做法：

把所有点以深度为第一关键字，$dfs$序为第二关键字排序。然后$k$代兄弟一定是在一段连续的区间中的，这样就转化成区间问题了。

这题的话，排好序后，对询问点二分出左右边界，就可以找到它的所有$k$代兄弟了。
（加强版是求 $k$代兄弟中的第$k$大，用主席树维护一下区间的就好了）

代码：

#include<bits/stdc++.h>
using namespace std;

int n;
vector<int>v[100005];
int id[100005],idx,dep[100005],fa[100005][20];
void dfs(int x,int pre)
{
    if(x==0)dep[x]=0,fa[x][0]=0;
    else dep[x]=dep[pre]+1,fa[x][0]=pre;
    if(x!=0)id[x]=++idx;
    for(auto to:v[x])
    {
        if(to!=pre)
        {
            dfs(to,x);
        }
    }
}

bool cmp(int a,int b)
{
    if(dep[a]!=dep[b])return dep[a]<dep[b];
    else return id[a]<id[b];
}

int getK(int x,int k)
{
    for(int i=0;i<=18;i++)if((k>>i)&1)x=fa[x][i];
    return x;
}

int idd[100005];
int wei[100005];
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        idd[i]=i;
        int x;
        scanf("%d",&x);
        v[x].push_back(i);
        v[i].push_back(x);
    }
    dfs(0,-1);
    for(int i=1;i<=18;i++)
    {
        for(int j=1;j<=n;j++)fa[j][i]=fa[fa[j][i-1]][i-1];
    }
    sort(idd+1,idd+n+1,cmp);
    for(int i=1;i<=n;i++)wei[idd[i]]=i;
    int m;
    scanf("%d",&m);
    while(m--)
    {
        int x,k;
        scanf("%d%d",&x,&k);
        int f=getK(x,k);
        if(f==0){printf("%d\n",0);continue;}
        int led,red;
        int l=1,r=wei[x];
        while(l<r)
        {
            int p=(l+r)/2;
            if(getK(idd[p],k)==f)r=p;
            else l=p+1;
        }
        led=l;

        l=wei[x],r=n;
        while(l<r)
        {
            int p=(l+r+1)/2;
            if(getK(idd[p],k)==f)l=p;
            else r=p-1;
        }
        red=l;

        printf("%d\n",red-led+1-1);
    }
    return 0;
}