Codeforce 208E——Blood Cousins（dfs序+倍增寻找父节点+二分）

题目链接：https://codeforces.com/problemset/problem/208/E

题目大意：

给你一个有根树，树的根为0节点，给你m个询问，每次询问会询问，对于v这个节点的第p个祖先vp，有多少个节点的第p个祖先也是vp？如果是0那就输出0。

思路：

这题让我理解了如果求树上共享祖先的同一深度的节点数

首先，我们必须得找到一个节点vp，它是v的第p个祖先，那么这个操作可以用倍增完成，道理和LCA是一样的，这里给出模板代码：

int p[maxn][20];//表示点i的第2^i个祖先是哪个
int d[maxn];//深度
void dfs(int u,int fa){
    d[u]=d[fa]+1;
    p[u][0]=fa;
    for(int i=1;i<POW;i++) p[u][i]=p[p[u][i-1]][i-1];
    int sz=G[u].size();
    for(int i=0;i<sz;i++){
        int v=G[u][i];
        if(v==fa) continue;
        dfs(v,u);
    }
}
//求a和b的LCA
int lca( int a, int b ){
    if( d[a] > d[b] ) a ^= b, b ^= a, a ^= b;
    if( d[a] < d[b] ){
        int del = d[b] - d[a];
        for( int i = 0; i < POW; i++ ) if(del&(1<<i)) b=p[b][i];
    }
    if( a != b ){
        for( int i = POW-1; i >= 0; i-- )
            if( p[a][i] != p[b][i] )
                 a = p[a][i] , b = p[b][i];
        a = p[a][0], b = p[b][0];
    }
    return a;
}
//寻找v的第pp个祖先
int Find(int v,int pp){
    for(int i=0;i<18;i++){
        if(pp&(1<<i))v=p[v][i];
    }
    return v;
}

用这个Find函数就可找到父节点了。

然后，我们对每个节点，记录一下这个节点的三个信息：节点深度depth[u]，节点入度dfs序in[u]，节点出度dfs序out[u]

我们设它的第p的父节点为vp

我们用一个vector记录每个深度的入度和出度，然后对于每个v，我们lower_bound查找相同深度的第一个大于等于它的父节点vp的入度的位置，再用lower_bound查找相同深度第一个大于等于它的父节点vp的出度的位置，两者形成一个区间，这个区间的长度，就是vp的子树的大小，然后你是在同一个深度查找的，所以现在的长度，就是vp的子树且和v在同一个深度的节点数个数

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

const int maxn=1e5+10;
vector<int>G[maxn];
vector<int>g[maxn];
void init(int n){
    for(int i=0;i<=n;i++){
        G[i].clear();
        g[i].clear();
    }
}
//用于求解LCA和保存公共祖先的dfs模板
int p[maxn][20];//表示点i的第2^i个祖先是哪个
int in[maxn];
int out[maxn];
int dep[maxn];
int dfn,depth;
const int POW =18;
int vis[maxn];

void dfs(int u,int fa){
    in[u]=++dfn;
    vis[u]=1;
    dep[u]=dep[fa]+1;
    depth=max(depth,dep[u]);
    g[dep[u]].push_back(dfn);//记录同一个深度的dfn
    p[u][0]=fa;//u的第一个父亲为fa
    for(int i=1;i<18;i++)p[u][i]=p[p[u][i-1]][i-1];//寻找第2^i的父亲
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i];
        if(vis[v])continue;
        dfs(v,u);
    }
    out[u]=++dfn;//出的dfn
}
//寻找v的第pp个祖先
int Find(int v,int pp){
    for(int i=0;i<18;i++){
        if(pp&(1<<i))v=p[v][i];
    }
    return v;
}

int ans[maxn];
signed main(){
    int n;
    while(scanf("%d",&n)!=EOF){
        init(n);
        dfn=0;//dfs序
        memset(p,0,sizeof(p));
        dep[0]=0;
        depth=0;
        for(int i=1;i<=n;i++){
            int u;
            scanf("%d",&u);
            G[u].push_back(i);
            G[i].push_back(u);
        }
        int root=0;
        memset(vis,0,sizeof(vis));
        dfs(0,0);
        int T;
        for(int i=0;i<=n;i++){
            cout<<i<<" in "<<in[i]<<endl;
            cout<<i<<" out "<<out[i]<<endl;
        }
        scanf("%d",&T);
        int x=T;
        int inx=0;
        while(T--){
            int v,pp;
            scanf("%d%d",&v,&pp);
            int vp=Find(v,pp);//祖先
            //cout<<"debug "<<vp<<endl;
            if(vp==0){
                ans[++inx]=0;
            }
            else{
                int x1=lower_bound(g[dep[v]].begin(),g[dep[v]].end(),in[vp])-g[dep[v]].begin();
                int y1=lower_bound(g[dep[v]].begin(),g[dep[v]].end(),out[vp])-g[dep[v]].begin();
                ans[++inx]=y1-x1-1;
            }
        }
        for(int i=1;i<=inx;i++)printf("%d ",ans[i]);
        puts("");
    }

}
/*
0 1 2 2 1 5 5 0 8 9 9 8 11 11
*/