节点的最近公共祖先（LCA倍增法）

问题描述

树是一种很常见的数据结构。现在蒜头君面临一个问题，在一个有 n 个节点的树上，节点编号分别是1…n。蒜头想知道一些节点之间的最近公共祖先是那些节点。
输入格式
第一行输入一个整数 n（2≤n≤10,000），表示树上有 n 个节点。
接下来的 n−1 行，每行输入俩个整数 a，b（1≤a,b≤n）代表节点 a，b 之间有一条 a 到 b 边，a 是 b 的父亲。
接下来输入一个整数 q，代表蒜头君的 q 次提问。（1≤q≤1,000）
接下来的 q 行，每行输入俩个整数 c，d（1≤c,d≤n）代表询问 c，d 俩个节点的最近公共祖先。
输出格式
对于每次询问，输出公共祖先的节点编号，占一行。
样例输入
5
1 2
2 3
1 4
2 5
2
3 4
3 5
样例输出
1
2

ac代码

#include <iostream>
#include <cstring>
#include <cstring>
#include <cstdio>
using namespace std;

const int MAXN=1e4+1;//最大的点数
struct edge{
    int v,next;//边的终点，同起点的上一条边的序号
    edge(){}
    edge(int _v,int _next):v(_v),next(_next){}//构造函数
}e[MAXN];
int p[MAXN],vst[MAXN],d[MAXN],fa[MAXN][20];//存储边的序号的头指针,标记数组,存储深度,fa[i][j]为i往上走(2^j)层的序号
void dfs(int u){//预处理深度和各节点的父亲节点
    for(int i=p[u];i+1;i=e[i].next){
        if(d[e[i].v]==-1){//该起点还未赋值
            d[e[i].v]=d[u]+1;//儿子等于父亲节点深度+1
            fa[e[i].v][0]=u;//记录父亲节点
            dfs(e[i].v);//递归更新深度和父亲
        }
    }
}
int lca(int x,int y){
    if(d[x]<d[y]){//始终让x节点为深一点的节点
        swap(x,y);
    }
    int i,j;
    for(i=0;(1<<i)<=d[x];++i);//找出（2^i<=n），找出i的最大值，保证不会超出树
    --i;
    //深的节点先往上走，走到和y节点同一层
    for(j=i;j>=0;--j){//先走大步
        if(d[x]-(1<<j)>=d[y]){//若往上走了（2^j）个数，任然深度比较大，
            x=fa[x][j];//就走之后的点为起点继续走
        }
    }
    if(x==y)return x;//若走到同一层时，两者相等，则y就是原先节点的父节点
    for(j=i;j>=0;--j){
//        if(x!=y){//自己第一个次是这样写的，（有bug），
//            x=fa[x][j];
//            y=fa[y][j];
//        }
        if(fa[x][j]!=fa[y][j]){//这样走下去，会走到深度最小的不等于两者不相同的点
            x=fa[x][j];
            y=fa[y][j];
        }
    }
    return fa[x][0];//此时x和y的父节点就是最近公共祖先
}

int main()
{
    int n;
    scanf("%d",&n);
    memset(vst,0,sizeof(vst));
    memset(p,-1,sizeof(p));
    for(int i=0;i<n-1;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        e[i]=edge(v,p[u]);
        p[u]=i;
        vst[v]=1;
    }
    int root;
    for(int i=1;i<=n;++i){
        if(!vst[i]){
            root=i;
            break;
        }
    }
    memset(d,-1,sizeof(d));
    dfs(root);
    for(int level=1;(1<<level)<=n;++level){
        for(int i=1;i<=n;++i){
            fa[i][level]=fa[fa[i][level-1]][level-1];
        }
    }
    int q;
    scanf("%d",&q);
    while(q--){
        int c,d;
        scanf("%d%d",&c,&d);
        printf("%d\n",lca(c,d));
    }
}