algorithm-最近公共祖先-LCA
[toc]
✨你好啊,我是“ 罗师傅”,是一名程序猿哦。
🌍主页链接:楚门的世界 - 一个热爱学习和运动的程序猿
☀️博文主更方向为:分享自己的快乐
❤️一个“不想让我曾没有做好的也成为你的遗憾”的博主。
💪很高兴与你相遇,一起加油!
前言
最近公共祖先(Least Common Ancestors, LCA)
参考博文:
洛谷题址:P3379 【模板】最近公共祖先(LCA
树上的倍增
推导公式:
1
2 for (int i = 1; (1 << i) <= d[u]; i++)
p[u][i] = p[p[u][i - 1]][i - 1];AcCode
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73 //
// Created by luozongwei on 2023/7/14.
//
using namespace std;
const int maxn = 500000 + 2;
int n, m, s;
int k = 0;
//head数组就是链接表标配了吧?d存的是深度(deep),p[i][j]存的[i]向上走2的j次方那么长的路径
int head[maxn], d[maxn], p[maxn][21];
struct node {
int v, next;
} e[maxn * 2];//存树
//加边函数
void add(int u, int v) {
e[k].v = v;
e[k].next = head[u];
head[u] = k++;
}
//首先进行的预处理,将所有点的deep和p的初始值dfs出来
void dfs(int u, int fa) {
d[u] = d[fa] + 1;
p[u][0] = fa;
for (int i = 1; (1 << i) <= d[u]; i++)
p[u][i] = p[p[u][i - 1]][i - 1];
for (int i = head[u]; i != -1; i = e[i].next) {
int v = e[i].v;
if (v != fa)
dfs(v, u);
}
}
//非常标准的lca查找
int lca(int a, int b) {
//保证a是在b结点上方,即a的深度小于b的深度
if (d[a] > d[b])
swap(a, b);
//先把b移到和a同一个深度
for (int i = 20; i >= 0; i--)
if (d[a] <= d[b] - (1 << i))
b = p[b][i];
//特判,如果b上来和就和a一样了,那就可以直接返回答案了
if (a == b)
return a;
for (int i = 20; i >= 0; i--) {
if (p[a][i] == p[b][i])
continue;
else
a = p[a][i], b = p[b][i]; //A和B一起上移
}
return p[a][0];
// 找出最后a值的数字
}
int main() {
memset(head, -1, sizeof(head));
int a, b;
scanf("%d%d%d", &n, &m, &s);
for (int i = 1; i < n; i++) {
scanf("%d%d", &a, &b);
add(a, b);
add(b, a); //无向图,要加两次
}
dfs(s, 0);
for (int i = 1; i <= m; i++) {
scanf("%d%d", &a, &b);
printf("%d\n", lca(a, b));
}
return 0;
}
❤️❤️❤️忙碌的敲代码也不要忘了浪漫鸭!
果然算法还是太难了,呜呜呜~~💦💦💦。
本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 楚门的世界!
评论