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

题目链接

题目描述

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

输入格式

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

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

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

输出格式

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

输入输出样例

输入

5 5 4
3 1
2 4
5 1
1 4
2 4
3 2
3 5
1 2
4 5

输出

4
4
1
4
4

#include<bits/stdc++.h>
using namespace std;
const int maxn = 5e5 + 5;
int h[maxn], etot;
int up[maxn][30], dep[maxn];

struct edge{
	int v, next;
}e[maxn * 2];

void add_edge(int x, int y)		// 添加边 
{
	e[etot].v = y;
	e[etot].next = h[x];
	h[x] = etot++;
}

void dfs(int u, int fa)		// 获得每个节点的父节点信息 
{
	dep[u] = dep[fa] + 1;
	up[u][0] = fa;
	for(int i = 1; (1 << i) <= dep[u]; i++)
		up[u][i] = up[up[u][i-1]][i-1];
	for(int i = h[u]; i != -1; i = e[i].next){
		if(e[i].v != fa){
			dfs(e[i].v, u);
		}
	}
}

int lca(int u,int v)
{
	if(dep[u] < dep[v])		// 令 u 是深度大的点 
		swap(u, v);
	if(dep[u] != dep[v]){		// 若深度不同，则让 u 往上爬到和 v 同一深度 
		for(int k = 20; k >= 0; k--){
			if(dep[up[u][k]] >= dep[v])
				u = up[u][k];
		}
	}
	if(u == v)		// 若此时 u 已经等于 v，就可以直接返回了 
		return u;
	for(int k = 20; k >= 0; k--){
		if(up[u][k] != up[v][k]){		// u 和 v 一起向上爬 
			u = up[u][k];
			v = up[v][k];
		}
	}
	return up[u][0];
}

int main()
{
	int n, m, s;
	cin >> n >> m >> s;
	memset(h, -1, sizeof(h));
	for(int i = 1; i <= n-1; i++){
		int x, y;
		cin >> x >>y;
		add_edge(x, y);
		add_edge(y, x);
	}
	dfs(s, 0);
	for(int i = 1; i <= m; i++){
		int x, y;
		cin >> x >> y;
		cout << lca(x, y) <<endl;
	}
	return 0;
}