算法提高-最近公共祖先

AcWing 1172. 祖孙询问

很裸的一道最近公共祖先的题目

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

const int N = 4e4 + 10;
int h[N], e[2 * N], ne[2 * N], idx;//无向图确实要*2，但是为啥不是N*N，
            //我感觉是题目给的“给定一棵包含 n个节点的有根无向树，节点编号互不相同，但不一定是 1∼n。
            //1≤n,m≤4×104
            //1≤每个节点的编号≤4×104
            //所以一共列n个，那肯定边数最多是n,然后无向图两个方向都要，就*2；
int q[N], depth[N], fa[N][16];
int n, m;
int root;

void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

void bfs()
{
    memset(depth, 0x3f, sizeof depth);
    int hh = 0, tt = -1;
    q[++tt] = root;
    depth[root] = 1;
    depth[0] = 0;//相当于哨兵
    while (hh <= tt)
    {
        int t = q[hh++];
        for (int i = h[t]; ~i; i = ne[i])
        {
            int j = e[i];
            if (depth[j] > depth[t] + 1) 
            {
                depth[j] = depth[t] + 1;
                q[++tt] = 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);//保证a的深度最深
    
    for (int k = 15; k >= 0; --k)
    {
        if (depth[fa[a][k]] >= depth[b])
        {
            a = fa[a][k];
        }
    }
    if (a == b) return b;//（二进制可以拼凑成任何>=1的数）经过二进制拼凑后，此时a一定和b在同一深度，如果a此时就是节点b，那么b此时就是最近公共祖先（别的题解里面是return a,其实此时a和b的值是一样的）
    for (int k = 15; k >= 0; -- k)
    {
        if (fa[a][k] != fa[b][k])//a和b在同一高度后，如果二者跳完不会达到同一节点就一直向上跳，这样比较好收集结果，保证二者最后是在最近公共祖先的下面一层
        {
            a = fa[a][k]; b = fa[b][k];
        }
    }
    
    return fa[a][0];//此时二者在最近公共祖先的下面一层，返回fa[b][0]一样的
}
void solve()
{
    cin >> n;
    memset(h, -1, sizeof h);//忘了这一步了，TLE了
    for (int i = 0; i < n; ++ i)
    {
        int a, b; cin >> a >> b;
        if (b == -1) root = a;
        else add(a, b), add(b, a);
    }
    bfs();
    scanf("%d", &m);
    while (m -- )
    {
        int a, b;
        scanf("%d%d", &a, &b);
        int p = lca(a, b);
        if (p == a) puts("1");
        else if (p == b) puts("2");
        else puts("0");
    }
}
int main()
{
    solve();
    return 0;
}