最近公共祖先（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

说明/提示

对于 30%30% 的数据，�≤10N≤10，�≤10M≤10。

对于 70%70% 的数据，�≤10000N≤10000，�≤10000M≤10000。

对于 100%100% 的数据，1≤�,�≤5000001≤N,M≤500000，1≤�,�,�,�≤�1≤x,y,a,b≤N，不保证 �≠�a=b。

样例说明：

该树结构如下：

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

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

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

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

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

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

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

#include<iostream>
#include<vector>
#include<math.h>
using namespace std;
const int N=500005;
int f[N][21];//ST表
int dep[N];//深度 
vector<int> e[N];
int n,m,s;//n为结点个数，m为询问次数，s为根节点的序号

void dfs(int x,int father){//构造ST表 
	dep[x]=dep[father]+1;
	f[x][0]=father;
	for(int i=1;i<21;i++){
		f[x][i]=f[f[x][i-1]][i-1]; 
	}
	for(int i=0;i<e[x].size();i++){
		if(e[x][i]!=father) dfs(e[x][i],x); 
	}
}
int LCA(int u,int v){
	if(dep[u]<dep[v]) swap(u,v);
	//跳到同一层 
	while(dep[u]>dep[v]){
		u=f[u][(int)log2(dep[u]-dep[v])];
	}
	if(u==v) return u;
	//跳到LCA的下一层
	for(int i=log2(dep[u]);i>=0;i--){
		if(f[u][i]!=f[v][i]){
			u=f[u][i];
			v=f[v][i];
		}
	} 
	return f[u][0];//再跳一步 
}
int main(){
	cin>>n>>m>>s;
	for(int i=1;i<=n-1;i++){
		int a,b;
		cin>>a>>b;
		e[a].push_back(b);
		e[b].push_back(a);
	} 
	dep[0]=0; 
	dfs(s,0);
	while(m--){
		int u,v;
		cin>>u>>v;
		cout<<LCA(u,v)<<endl;
	}
	return 0;
} 

思路：先求出ST表，然后将较深的点跳到和另一个点同样的高度，然后两个点一起跳，直到跳到最近公共祖先的下一个，再往上跳一步即可。（两个点一起跳的时候可能跳到较远的祖先，所以要循环判断，我们的目标不是跳到最近祖先，而是最近祖先的下一个）