[CodeForces - 208E] Blood Cousins(k代兄弟问题)

题目链接
加强版例题(还没写)

题意:

给定一棵 n n n个结点的树。给出 m m m次询问,每次询问给出 x x x k k k,问 x x x有多少个 k k k代兄弟。(两个点互为 k k k代兄弟的定义是他们的 k k k级祖先相同)
1 ≤ n , m ≤ 1 0 5 1\leq n,m \leq 10^{5} 1n,m105

做法:

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

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

代码:

#include
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;
}

你可能感兴趣的:(树学)