【模板】最近公共祖先（LCA）

【模板】最近公共祖先（LCA）

题目描述

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

输入格式

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

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

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

输出格式

输出包含 $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\%$ 的数据，$N\le 10$，$M\le 10$。

对于 $70\%$ 的数据，$N\le 10000$，$M\le 10000$。

对于 $100\%$ 的数据，$1\le N,M\le 500000$，$1\le x,y,a,b\le N$，不保证 $a\ne b$。

样例说明：

该树结构如下：

第一次询问：$2,4$ 的最近公共祖先，故为 $4$。

第二次询问：$3,2$ 的最近公共祖先，故为 $4$。

第三次询问：$3,5$ 的最近公共祖先，故为 $1$。

第四次询问：$1,2$ 的最近公共祖先，故为 $4$。

第五次询问：$4,5$ 的最近公共祖先，故为 $4$。

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

#include <iostream>
#include <algorithm>
using namespace std;
int n, m, s, u, v, cnt;
int h[501000], to[1001000], nxt[1001000], dep[501000], lg[501000];
int f[501000][25];
inline void add_edge(const int& u, const int& v) {
	to[++cnt] = v;
	nxt[cnt] = h[u];
	h[u] = cnt;
}
void dfs(const int& x) {
	for (register int i = 0; f[x][i]; ++i) f[x][i + 1] = f[f[x][i]][i];
	for (register int i = h[x]; i; i = nxt[i])
		if (to[i] ^ f[x][0]) {
			dep[to[i]] = dep[x] + 1;
			f[to[i]][0] = x;
			dfs(to[i]);
		}
}
inline int lca(int x, int y) {
	if (dep[x] < dep[y]) swap(x, y);
	while (dep[x] > dep[y]) x = f[x][lg[dep[x] - dep[y]]];
	if (x == y) return x;
	for (register int i = lg[dep[x]] + 1; i--;)
		if (f[x][i] ^ f[y][i])
			x = f[x][i], y = f[y][i];
	return f[x][0];
}
int main() {
	scanf("%d%d%d", &n, &m, &s);
	for (register int i = 2; i <= n; ++i) {
		lg[i] = lg[i >> 1] + 1;
		scanf("%d%d", &u, &v);
		add_edge(u, v);
		add_edge(v, u);
	}
	dfs(s);
	for (register int i = 1; i <= m; ++i) {
		scanf("%d%d", &u, &v);
		printf("%d\n", lca(u, v));
	}
}