祖孙询问(AcWing)

给定一棵包含 n个节点的有根无向树，节点编号互不相同，但不一定是 1∼n。

有 m个询问，每个询问给出了一对节点的编号 x和 y，询问 x 与 y 的祖孙关系。

输入格式

输入第一行包括一个整数n,表示节点个数；

接下来 n行 每行一对整数 a和 b，表示 a和 b之间有一条无向边。如果 b是 −1，那么 a就是树的根；

第 n+2 行是一个整数 m 表示询问个数；

接下来 m行，每行两个不同的正整数 x 和 y，表示一个询问。

输出格式

对于每一个询问，若 x 是 y 的祖先则输出 1，若 y 是 x 的祖先则输出 2，否则输出 0。

数据范围

每个节点的编号

思路

最基本的LCA（最近公共祖先）问题

这里采用倍增的做法，记录一个标准化的模板

#include<bits/stdc++.h>
using namespace std;
const int N=4e4+10;
vector<int> edge[N];
int f[N][32];
int dep[N];
map<int,int>mp;
int n,m,root;
void dfs(int u,int fa){
    f[u][0]=fa;
    for(int i=1;i<=20;i++){
        f[u][i]=f[f[u][i-1]][i-1];
    }
    for(int i=0;i<edge[u].size();i++){
        int v=edge[u][i];
        if(v==fa) continue;
        dep[v]=dep[u]+1;
        dfs(v,u);
    }
    return ;
}
int lca(int x,int y){
    if(dep[x]>dep[y]) swap(x,y);
    int dd=dep[y]-dep[x];
    for(int i=0;i<=20;i++){
        if((1<<i)&dd) y=f[y][i];
    }
    if(x==y) return x;
    for(int i=20;i>=0;i--){
        if(f[x][i]!=f[y][i]){
            x=f[x][i];
            y=f[y][i];
        }
    }
    return f[x][0];
}
int main(void){

    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        if(b==-1){
            root=a;
        }
        else{
            edge[a].push_back(b);
            edge[b].push_back(a);
        }
    }
    scanf("%d",&m);
    dep[root]=0;
    dfs(root,0);
    for(int i=1;i<=m;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        int u=lca(x,y);
        if(u==x){
            printf("1\n");
        }
        else if(u==y){
            printf("2\n");
        }
        else printf("0\n");
    }



    return 0;
}