洛谷 P3379 【模板】最近公共祖先(LCA) 做题反思(2024.3.21)

【模板】最近公共祖先(LCA)

题目描述

如题,给定一棵有根多叉树,请求出指定两个点直接最近的公共祖先。

输入格式

第一行包含三个正整数 N , M , S N,M,S N,M,S,分别表示树的结点个数、询问的个数和树根结点的序号。

接下来 N − 1 N-1 N1 行每行包含两个正整数 x , y x, y x,y,表示 x x x 结点和 y y y 结点之间有一条直接连接的边(数据保证可以构成树)。

接下来 M M M 行每行包含两个正整数 a , b a, b a,b,表示询问 a a a 结点和 b b b 结点的最近公共祖先。

输出格式

输出包含 M M M 行,每行包含一个正整数,依次为每一个询问的结果。

样例 #1

样例输入 #1

5 5 4
3 1
2 4
5 1
1 4
2 4
3 2
3 5
1 2
4 5

样例输出 #1

4
4
1
4
4

提示

对于 30 % 30\% 30% 的数据, N ≤ 10 N\leq 10 N10 M ≤ 10 M\leq 10 M10

对于 70 % 70\% 70% 的数据, N ≤ 10000 N\leq 10000 N10000 M ≤ 10000 M\leq 10000 M10000

对于 100 % 100\% 100% 的数据, 1 ≤ N , M ≤ 500000 1 \leq N,M\leq 500000 1N,M500000 1 ≤ x , y , a , b ≤ N 1 \leq x, y,a ,b \leq N 1x,y,a,bN不保证 a ≠ b a \neq b a=b

样例说明:

该树结构如下:

第一次询问: 2 , 4 2, 4 2,4 的最近公共祖先,故为 4 4 4

第二次询问: 3 , 2 3, 2 3,2 的最近公共祖先,故为 4 4 4

第三次询问: 3 , 5 3, 5 3,5 的最近公共祖先,故为 1 1 1

第四次询问: 1 , 2 1, 2 1,2 的最近公共祖先,故为 4 4 4

第五次询问: 4 , 5 4, 5 4,5 的最近公共祖先,故为 4 4 4

故输出依次为 4 , 4 , 1 , 4 , 4 4, 4, 1, 4, 4 4,4,1,4,4

2021/10/4 数据更新 @fstqwq:应要求加了两组数据卡掉了暴力跳。

最近公共祖先(LCA)的概念
两个节点的最近公共祖先(LCA)就是这两个点的公共祖先里面,离他们最近的那个。
如图所示:
在这里插入图片描述
例如:
9和8的祖先节点有1、5、6,但是离9和8最近的公共祖先是节点6

倍增算法:是最经典的求LCA算法
∙ \bullet dep[u]是存u点的深度;
∙ \bullet fa[u][i]存从u点向上跳 2 i 2^i 2i层的祖先节点。(i=0,1,2,3,…);

例如:
从节点9向上跳 2 0 2^0 20层的祖先是3,跳 2 1 2^1 21层的祖先是6,跳 2 2 2^2 22层的祖先是1,跳 2 3 2^3 23层的祖先是0(即越界)。

1、dfs一遍,创建ST表
倍增递推,fa[u][i]=fa[fa[u][i-1]][i-1]
在这里插入图片描述
假设u是节点9,则层数增加两倍到倒数第四层,再除以2到倒数第二层,找到节点6,依次递归到节点1。

如表所示:

012319
100000
510000
251000
651000
365000
936100

2、利用ST表求LCA
(1)第一阶段,将u,v跳到同一层
设u,v两点的深度之差为y,将y进行二进制拆分,可以将y次游标跳跃优化为“y的二进制表示所有含1的个数”次游标跳跃,一定能跳到同一层。
例如,y=1019(0…01111111011)=512+256+…+8+2+1,不越界则跳,共跳9次到达。
(2)第二阶段,将u,v一起跳到LCA的下一层
从最大的i开始循环尝试,一直尝试到0,最后游标u,v一定能停在LCA的下一层。
两游标会跳512+256+…+8+2=1018层,共跳8次到达LCA的下一层。

代码示例如下:

#include<iostream>
#include<algorithm>
using namespace std;
const int N = 5e5 + 10;
int n, m, s;
int f[N][22], dep[N];
int e[2 * N], ne[2 * N], h[N], idx;
void add(int a, int b) {
	e[++idx] = b, ne[idx] = h[a], h[a] = idx;
}
void dfs(int u, int fa) {
	dep[u] = dep[fa] + 1;//计算节点u的深度
	f[u][0] = fa;//找节点u的父节点
	for (int i = 1; i <= 20; i++) {//这里从1循环到20的原因是因为n的最大范围是5e5+10,所以2的20次方是一百万,远远大于5e5+10,这样保证数据不会溢出
		f[u][i] = f[f[u][i - 1]][i - 1];//找到所有的公共祖先
	}
	for (int i = h[u]; i; i = ne[i]) {
		int j = e[i];
		if (j != fa)dfs(j, u);//如果没有找到父节点就继续搜
	}
}
int lca(int u, int v) {
	if (dep[u] < dep[v])swap(u, v);//先跳到同一层
	for (int i = 20; i >= 0; i--) {
		if (dep[f[u][i]] >= dep[v]) {//将节点u继续往上跳
			u = f[u][i];
		}
	}
	if (u == v)return v;//找到最近公共祖先
	for (int i = 20; i >= 0; i--) {
		if (f[u][i] != f[v][i]) {//如果没有找到最近公共祖先
			u = f[u][i];
			v = f[v][i];
		}
	}
	return f[u][0];
}
int main() {
	scanf("%d%d%d", &n, &m, &s);
	for (int i = 1; i < n; i++) {
		int x, y;
		scanf("%d%d", &x, &y);
		add(x, y);
		add(y, x);
	}
	dfs(s, 0);
	while (m--) {
		int a, b;
		scanf("%d%d", &a, &b);
		printf("%d\n", lca(a, b));
	}
	return 0;
}
  • 27
    点赞
  • 24
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

不会敲代码的狗

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值