倍增求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$。

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

#include <bits/stdc++.h>
using namespace std;
const int N = 500005;
int de[N];//记录每个点的深度 
vector<int> v[N];
int f[N][25];//记录i点向上走2^K次方的祖先 
int n, m, root;
void bfs(int root) {
	memset(de, 0x3f3f3f, sizeof de);
	queue<int> q;
	q.push(root);
	de[0] = 0;//哨兵 
	de[root] = 1;
	while (q.size()) {
		int t = q.front();
		q.pop();
		for (int i = 0; i < v[t].size(); i++) {
			int j = v[t][i];
			if (de[j] > de[t] + 1) {
				q.push(j);
				de[j] = de[t] + 1;
				f[j][0] = t;
				for (int k = 1; k <= 22; k++) {
					f[j][k] = f[f[j][k - 1]][k - 1];
				}
			}
		}
	}
}
int lca(int a, int b) {
	if (de[a] < de[b])swap(a, b);
	for (int k = 22; k >= 0; k--) {
		if (de[f[a][k]] >=de[b])
			a = f[a][k];
	}
	if(a==b)return a;
	for(int k=22;k>=0;k--)
	{
		if(f[a][k]!=f[b][k])
		{
			a=f[a][k];
			b=f[b][k];
		}
	}
	return f[a][0];
}
int main() {
	int a, b;
	scanf("%d%d%d", &n, &m, &root);
	for (int i = 1; i < n; i++) {
		scanf("%d%d", &a, &b);
		v[a].push_back(b), v[b].push_back(a);
	}
	bfs(root);
	while (m--) {
		scanf("%d%d", &a, &b);
		printf("%d\n",lca(a,b));
	}
}