【Jeoy‘ s daily 】AcWing 1172. 祖孙询问 简单LCA，倍增

AcWing 1172 祖孙询问
题面

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

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

输入格式
输入第一行包括一个整数 表示节点个数；

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

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

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

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

数据范围
1≤n,m≤4×104,
1≤每个节点的编号≤4×104
输入样例：
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

*模板题，直接套倍增模板，但是1171套模板就写不出来，我还是太菜了，看题解是tarjan，不会就是说…
*
原题链接

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e6;

LL idx, h[N * 2], e[N * 2], ne[N * 2], w[N *2];
LL n, m, depth[N * 2], f[N][16], p[N], len[N][16];

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

void bfs ( int r ) {
    memset( depth, 0x3f, sizeof depth );
    depth[0] = 0; depth[r] = 1;
    LL hh = 0, tt = 0;
    p[0] = r;
    while ( hh <= tt ) {
        int t = p[hh ++ ];
     //   cout << t << endl;
        for ( int i = h[t]; ~i; i = ne[i] ) {
            LL s = e[i];
            if ( depth[s] > depth[t] + 1 ) {
                depth[s] = depth[t] + 1;
                f[s][0] = t, len[s][0] = w[i];
                for ( int i = 1; i <= 15; i++ ) {
                    f[s][i] = f[f[s][i - 1]][i - 1];
                }
                p[++tt] = s;
            }
        }
    }

}

LL lca ( int a, int b ) {
    if ( depth[b] > depth[a] ) swap( a, b );
    for ( int i = 15; i >= 0; i -- ) {
        if ( depth[f[a][i]] >= depth[b] ) {
            a = f[a][i];
        }
    }
    if ( a == b ) return a;
    for ( int i = 15; i >= 0; i -- ) {
        if ( f[a][i] != f[b][i] ) {
            a = f[a][i],b = f[b][i];
        }
    }
    return f[a][0];
}

int main () {
    memset( h, -1, sizeof h );
    scanf( "%lld", &n);
    int root = 0;
    for ( int i = 1; i <= n; i ++ ) {
        LL a, b;
        scanf( "%lld%lld", &a, &b );
      //  cout << a << b << endl;
        if ( b == -1 ) root = a;
        else {
            add( a, b );
            add( b, a );
        }
    }
   // cout << root << endl;
    bfs ( root );
    scanf( "%lld", &m );
    for ( int i = 1; i <= m; i++ ) {
        LL a, b;
        scanf( "%lld%lld" , &a, &b );
        int z = lca ( a, b );
        if ( z == a ) cout << "1" << endl;
        else if ( z == b ) cout << "2" << endl;
        else cout << "0" << endl;
    }
}