Acwing 1172. 祖孙询问（最近公共祖先——LCA模板题）

输入样例：

10
234 -1
12 234
13 234
14 234
15 234
16 234
17 234
18 234
19 234
233 19
5
234 233
233 12
233 13
233 15
233 19

输出样例：

1
0
0
0
2

 解题思路：标准的LCA模板题，首先我们要预处理出所有结点的深度，设根节点的深度为1，0号结点的深度为0，同时还需要求一下fa[i][j]，表示从i结点出发往上走2^j步到达的点的标号，具体的递推公式是：fa[i][j]=fa[fa[i][j-1][j-1]，算的时候j要从小到达来枚举。

然后进入求LCA环节，给定两个点，首先判断一下这两个点是深度相同，如果不同我们就要先让深度大的结点往上跳到和另一个结点深度相同处，然后两个结点一起往上跳，最终会跳到最近公共祖先的下一层，所以最后的答案就是fa[a][0]或者fa[b][0]，其中a,b是两个结点最后跳到的位置，这里需要注意的是 ，往上跳的时候，枚举跳跃的步伐时要从大到小进行枚举 ，如果按照当前的步伐往上跳后到达的点满足条件就直接跳到该点，否则继续枚举更小的步伐，所有步伐都枚举完之后，就跳到了终点。具体细节见代码注释。

上代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <queue>
using namespace std;
const int N=4e4+10;
int h[N],e[N<<1],ne[N<<1],idx,depth[N],fa[N][16],root;
int n,m;
void add(int u,int v)
{
    e[idx]=v;
    ne[idx]=h[u];
    h[u]=idx++;
}
void bfs(int root)
{
    memset(depth,0x3f3f3f3f,sizeof depth);
    depth[0]=0;//作为一个哨兵，用于处理跳出界的情况 
    depth[root]=1;
    queue<int>q;
    q.push(root);
    while(q.size())
    {
        int t=q.front();
        q.pop();
        for(int i=h[t];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(depth[j]>depth[t]+1)
            {
                depth[j]=depth[t]+1;
                q.push(j);
                fa[j][0]=t;
                for(int k=1;k<=15;k++)
                fa[j][k]=fa[fa[j][k-1]][k-1];
            }
        }
    }
}
int lca(int a,int b)
{
    if(depth[a]<depth[b])
    swap(a,b);
    //先让深度大的结点往上跳 
    for(int i=15;i>=0;i--)
    if(depth[fa[a][i]]>=depth[b])
    a=fa[a][i];
    if(a==b)
    return a;
    //两个点一块往上跳 
    for(int i=15;i>=0;i--)
    {
        if(fa[a][i]!=fa[b][i])//为了便于计算，我们最终要跳到最近公共祖先的下一个结点，所以跳跃之后两个点不能是一个点 
        {
            a=fa[a][i];
            b=fa[b][i];
        }
    }
    return fa[a][0];
}
int main()
{
    cin>>n;
    memset(h,-1,sizeof h);
    for(int i=1;i<=n;i++)
    {
        int u,v;
        cin>>u>>v;
        if(v==-1)
        root=u;
        else
        {
            add(u,v);
            add(v,u);   
        }
    }
    bfs(root);
    cin>>m;
    for(int i=1;i<=m;i++)
    {
        int a,b;
        cin>>a>>b;
        int t=lca(a,b);
        if(t==a)
        cout<<1<<endl;
        else if(t==b)
        cout<<2<<endl;
        else
        cout<<0<<endl;
    }
    return 0;
}