Luogu P3379 【模板】最近公共祖先(LCA) 重链剖分

文章介绍了一种使用重链剖分解决有根多叉树中两个节点最近公共祖先(LCA)问题的方法。通过树的深度优先遍历构建重链,对于同一重链上的节点,它们的LCA即为深度较小的节点;不同重链则通过不断回溯至相同重链找到LCA。该算法优化了查询复杂度。

题目描述

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

输入格式

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

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

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

输出格式

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

样例 #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≤10N\leq 10N10M≤10M\leq 10M10

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

对于 100%100\%100% 的数据,1≤N,M≤5000001 \leq N,M\leq 5000001N,M5000001≤x,y,a,b≤N1 \leq x, y,a ,b \leq N1x,y,a,bN不保证 a≠ba \neq ba=b

样例说明:

该树结构如下:

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

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

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

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

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

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

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

重链剖分 - AC

树链剖分分为重链剖分、长链剖分、实链剖分。
重链剖分将书划分为几条重链,对一条链统一处理,减小复杂度。节点的儿子中,子树最大的一个称为重儿子,连向重儿子的边称作重边。重边连成重链,轻边连接两条重链。显然,重链的顶端都是轻儿子。单独一个叶子轻儿子也被看作一条重链。
从一个节点往下走的路径上,最多有log2nlog_2nlog2n条重链。最坏情况是,每走一步,都经过一条轻边,也就是走到另一条重链上。从子树大小的角度考虑,那会丢掉重儿子的子树,以及其他轻儿子的子树,以及根节点。也就是,丢掉的很“重”,大于目前所有节点的一半。最后到了叶子节点。所以最多有log2nlog_2nlog2n条重链。重链剖分用巧妙的方式降低复杂度。
在这里插入图片描述


重链剖分可以解决LCA问题。
两个点如果在同一条重链上,LCA就是深度小的那个点。
否则,不断往上跳,使在同一条上。每次让顶端节点深度大的那个,跳到顶端节点的父节点。因为不断往上跳,跳到LCA所在重链,是一个深度不断减小的过程。

int n, m, s, siz[MAXN], son[MAXN], fa[MAXN], top[MAXN], dep[MAXN];
int js, head[MAXN], to[MAXN << 1], nxt[MAXN << 1];

void add(int u, int v) {
	to[js] = v; nxt[js] = head[u]; head[u] = js++;
}

void dfs1(int u, int par) {
	siz[u] = 1;
	for (int i = head[u]; ~i; i = nxt[i]) {
		int v = to[i];
		if (v == par) continue;
		dep[v] = dep[u] + 1;
		fa[v] = u;
		dfs1(v, u);
		siz[u] += siz[v];
		if (siz[v] > siz[son[u]]) son[u] = v;
	}
}
void dfs2(int u) {
	for (int i = head[u]; ~i; i = nxt[i]) {
		int v = to[i];
		if (v == fa[u]) continue;
		if (v == son[u]) top[v] = top[u];
		else top[v] = v;
		dfs2(v);
	}
}

int lca(int x, int y) {
	while (top[x] != top[y]) {
		if (dep[top[x]] > dep[top[y]]) x = fa[top[x]];
		else y = fa[top[y]];
	}
	return dep[x] > dep[y] ? y : x;
}

int main() {
	memset(head, -1, sizeof(head));
	
	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);
	}
	
	dfs1(s, 0);
	top[s] = s; dfs2(s);
	
	while (m--) {
		int a, b; scanf("%d%d", &a, &b);
		printf("%d\n", lca(a, b));
	}
	
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值