洛谷 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
    评论
二叉搜索树(Binary Search Tree,简称BST)是一种特殊的二叉树,它的每个节点的值都大于其左子树中的所有节点的值,且小于其右子树中的所有节点的值。最近公共祖先(Lowest Common Ancestor,简称LCA)是指在二叉树中,两个节点p和q的最近公共祖先节点。 对于给定的二叉搜索树,我们可以通过比较节点的值来确定最近公共祖先节点。具体步骤如下: 1. 从根节点开始遍历二叉搜索树。 2. 如果当前节点的值大于p和q的值,说明p和q都在当前节点的左子树中,因此继续遍历当前节点的左子树。 3. 如果当前节点的值小于p和q的值,说明p和q都在当前节点的右子树中,因此继续遍历当前节点的右子树。 4. 如果当前节点的值介于p和q的值之间,说明当前节点就是最近公共祖先节点。 以下是一个示例代码,演示了如何找到二叉搜索树的最近公共祖先节点: ```python class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right def lowestCommonAncestor(root, p, q): if root.val > p.val and root.val > q.val: return lowestCommonAncestor(root.left, p, q) elif root.val < p.val and root.val < q.val: return lowestCommonAncestor(root.right, p, q) else: return root # 创建一个二叉搜索树 root = TreeNode(6) root.left = TreeNode(2) root.right = TreeNode(8) root.left.left = TreeNode(0) root.left.right = TreeNode(4) root.right.left = TreeNode(7) root.right.right = TreeNode(9) root.left.right.left = TreeNode(3) root.left.right.right = TreeNode(5) # 找到节点3和节点5的最近公共祖先 p = TreeNode(3) q = TreeNode(5) lca = lowestCommonAncestor(root, p, q) print("最近公共祖先节点的值为:", lca.val) # 输出:2 ```

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

不会敲代码的狗

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

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

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

打赏作者

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

抵扣说明:

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

余额充值