【题目描述】
已知一棵 n 个节点的有根树。有 m 个询问,每个询问给出了一对节点的编号 x 和 y,询问 x 与 y 的祖孙关系。
【输入】
输入第一行包括一个整数 n 表示节点个数;
接下来 n 行每行一对整数对 a 和 b 表示 a 和 b 之间有连边。如果 b 是 −1,那么 a 就是树的根;
第 n+2 行是一个整数 m 表示询问个数;
接下来 m 行,每行两个正整数 x 和 y,表示一个询问。
【输出】
对于每一个询问,若 x 是 y 的祖先则输出 1,若 y 是 x 的祖先则输出 2,否则输出 0。
【输入样例】
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
【提示】
数据范围与提示:
对于 30% 的数据,1≤n,m≤103 ;
对于 100% 的数据,1≤n,m≤4×104 ,每个节点的编号都不超过 4×104 。
分析
- LCA的板题,使用倍增实现的;首先一个bfs函数进行数组 depth、fa的初始化,一个lca函数来求公共祖先,先一层for让两个点跳到同一层,再一层for同时往上跳到根的下一层,最后再跳一层返回;
- bfs: 先获取到队头u,然后遍历他的邻边v,初始化是自根向下的,所以fa[v][0]=u,说明儿子向上再跳一步就是根节点(就不用在设置fa[u][0]的值了,它是个根还往上跳啥),所以fa的初始化,是对fa[v][i]进行赋值的(自上而下),fa[v][k]可可以跳两次实现,先跳2^(k-1) 步到 fa[v][k-1],再跳2^(k-1) 步到达fa[v][k],利用的递推的思想(2^k-1 + 2^k-1 =>2^k);
- lca: 先判断a,b大小,保证a在b的下面;然后开始处理让他们跳到同一层,当然判断的就是depth,让a往上跳,k从20倒着往下找,不用怕越界(有守卫),至到for结束a,b在同一层;此时判断下a是否=b,等的话说明找到公共祖先了;然后处理a,b同时跳,只要fa[a][k] != fa[b][k](只要没跳到同一个结点上,就只管跳)就还继续跳,至到跳到他们最近公共祖先的下一层;最后再跳一层返回,从a向上跳一步、b向上跳一步都行;
- 关于哨兵的设置和作用:哨兵一共设置两份,depth[0] = 0和 fa[i,j]=0(fa默认就是0,然后把能到达的在bfs中初始化);哨兵的作用就是对跳出根节点的操作进行处理,如果a跳出去了,然后fa[a][k]=0,那么无论是depth[fa[a][k]]之间的判断,还是fa[a][k]的判断,如果跳出根节点,自然不满足当时的if条件;
- 吐槽下这个题的假数据范围,需要多开一点(多开一万还不行,那就直接N再乘10),不然会被卡RE、WA;下图为yxc的思路描述:
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 400010, M = 400010 * 2;
int n, m;
int h[N], e[N], ne[N], idx;
int q[N];
//fa[i,j]表示从i向上走 2^j 步能到的点;depth表示深度
int depth[N], fa[N][21];// 2^16 >40010
void add(int a, int b) {
e[++idx] = b, ne[idx] = h[a], h[a] = idx;
}
//初始化depth、fa
void bfs(int root) {
memset(depth, 0x3f, sizeof depth);
//哨兵:depth[0] = 0,f[i,j]默认就是0了
depth[0] = 0;
int hh = 0, tt = 0;
//第一个点——根节点
q[0] = root;
depth[root] = 1;
while (hh <= tt) {
int u = q[hh++];
for (int i = h[u]; ~i; i = ne[i]) {
int v = e[i];
//如果u,v相差不是一层,说明v还没被搜过,加队列
if (depth[v] > depth[u] + 1) {
depth[v] = depth[u] + 1;
q[++tt] = v;
//初始化fa,对v开始的进行处理(自上而下)
fa[v][0] = u;
for (int k = 1; k <= 20; ++k) {
//先跳2^(k-1)步到f[v][k-1],再跳2^(k-1)步
fa[v][k] = fa[fa[v][k - 1]][k - 1];
}
}
}
}
}
//求两个点的公共祖先
int lca(int a, int b) {
//a要在b下面
if (depth[a] < depth[b])
swap(a, b);
//1. a先跳到同一层(一个数肯定能通过二进制表示,这层for结束就到了同一层)
for (int k = 20; k >= 0; --k) {
//a先向上跳2^k步
//不用怕k=16或者20起始值太大,有哨兵,不怕他第一次跳太远,导致直接跳出根节点,如果跳出根节点,f[i][j]=0会出手的
//哨兵的作用:如果f[a][k]跳出了根节点,那么左边就是depth[0]=0,自然不满足当前if,i--找下一个i
if (depth[fa[a][k]] >= depth[b]) {//=时候,直接跳到同一层,后序就进不了if了
//a赋上新位置,继续往上跳
a = fa[a][k];
}
}
//找到公共祖先
if (a == b)
return a;
//2. a,b同时往上跳,至到他们最近公共祖先的下一层
for (int k = 20; k >= 0; --k) {
//说明还没跳到公共祖先(没跳到同一个结点,原目的也就不想让他们直接跳到公共祖先),就一直跳
//哨兵的作用:如果a跳出去了,b肯定也跳出去了,因为同一层开始跳的,然后fa[a][k]=0,0!=0自然不满足当前的if,i--
if (fa[a][k] != fa[b][k]) {
//更新这次跳到的新位置
a = fa[a][k];
b = fa[b][k];
}
}
//当前在公共祖先的下一层,再往上跳一层即可
return fa[a][0];//fa[b][0]也是可以的
}
int main() {
memset(h, -1, sizeof h);
scanf("%d", &n);
int root = 0; //根节点
for (int i = 0; i < n; ++i) {
int a, b;
scanf("%d%d", &a, &b);
if (b == -1)
root = a;
else
add(a, b), add(b, a);//无向边
}
bfs(root);
cin >> m;
while (m--) {
int x, y;
scanf("%d%d", &x, &y);
int p = lca(x, y);
if (p == x)
printf("1\n");
else if (p == y)
printf("2\n");
else
printf("0\n");
}
return 0;
}