72 lines
2.2 KiB
C++
72 lines
2.2 KiB
C++
#include <bits/stdc++.h>
|
||
using namespace std;
|
||
const int N = 40010, M = 80010;
|
||
|
||
//链式前向星
|
||
int e[M], h[N], idx, ne[M];
|
||
void add(int a, int b) {
|
||
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
|
||
}
|
||
int n;
|
||
int depth[N];
|
||
int f[N][16];
|
||
|
||
//树上深搜
|
||
void dfs(int u) {
|
||
for (int i = h[u]; ~i; i = ne[i]) {
|
||
int j = e[i];
|
||
if (!depth[j]) {
|
||
depth[j] = depth[u] + 1; //① 每个节点的深度
|
||
f[j][0] = u; //② j的父节点是u
|
||
dfs(j);
|
||
}
|
||
}
|
||
}
|
||
|
||
int lca(int a, int b) {
|
||
if (depth[a] < depth[b]) swap(a, b);
|
||
|
||
for (int k = 15; k >= 0; k--)
|
||
if (depth[f[a][k]] >= depth[b]) a = f[a][k];
|
||
|
||
if (a == b) return a;
|
||
return f[a][0];
|
||
}
|
||
|
||
int main() {
|
||
memset(h, -1, sizeof h); //初始化链式前向星
|
||
scanf("%d", &n); // n个顶点
|
||
|
||
int a, b, m, root;
|
||
while (n--) {
|
||
scanf("%d %d", &a, &b); //表示a和b之间有一条无向边。如果b是−1，那么a就是树的根
|
||
if (b == -1)
|
||
root = a;
|
||
else
|
||
add(a, b), add(b, a); //有根无向树
|
||
}
|
||
|
||
//从根开始进行处理，根的深度是1
|
||
depth[root] = 1;
|
||
dfs(root); //从根开始深搜，目标是 ①：给每个节点标识上深度 ②:给每个节点设置上打表f[i][0]的值，其实就是他爹是谁
|
||
|
||
// dfs只解决了2^0问题，其它的2^1,2^2,2^3,...,2^k,还需要再写循环，自己实现，这个顺序与bfs不一样，bfs根据深度枚举，遇到i时，它的依赖前序都已准备好， dfs需要全跑完再来一遍
|
||
// 这个枚举序挺烦人啊，dp枚举顺序是配合业务来的,从这个来看，固定好2^i，然后逐个枚举每个点号进行填充
|
||
// 所以，yxc大佬没有讲按dfs进行标识深度，而是采用了bfs进行标识，bfs是很好理解
|
||
for (int i = 1; i <= 15; i++)
|
||
for (int j = 1; j <= N; j++)
|
||
f[j][i] = f[f[j][i - 1]][i - 1];
|
||
|
||
scanf("%d", &m);
|
||
while (m--) {
|
||
scanf("%d %d", &a, &b);
|
||
int p = lca(a, b);
|
||
if (p == a)
|
||
puts("1");
|
||
else if (p == b)
|
||
puts("2");
|
||
else
|
||
puts("0");
|
||
}
|
||
return 0;
|
||
} |