二叉树：提高：最近公共祖先（LCA）-模板2

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

输入

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

接下来 kk 行每行包含两个正整数 x,yx,y，表示 xx 结点和 yy 结点之间有一条直接连接的边（数据保证可以构成树，可能有重路，读入时删除）。

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

输出

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

样例

输入

5 4 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% 的数据，N≤10N≤10，M≤10M≤10。

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

对于 100%100% 的数据，1≤N,M≤5000001≤N,M≤500000，1≤x,y,a,b≤N1≤x,y,a,b≤N，不保证 a≠ba≠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。

———————————————————————————————————————————

代码：

#include<bits/stdc++.h>
using namespace std;
int n,k,m,s;
int d[500002];
vector<int> a[500002];
int f[500002][21];
void dfs(int u)
{
	queue<int>q;
	q.push(u),d[u]=1;
	while(!q.empty())
	{
		int x=q.front();
		q.pop();
		for(int i=0;i<a[x].size();i++)
		{
			int y=a[x][i];
			if(d[y]) continue;
			d[y]=d[x]+1;
			f[y][0]=x;
			for(int j=1;j<=20;j++)
				f[y][j]=f[f[y][j-1]][j-1];
			q.push(y);
		}
	}
}
int lca(int a,int b)
{
	if(d[a]<d[b]) swap(a,b);
	for(int i=20;i>=0;i--)               //找到和b同一高度的节点 
		if(d[f[a][i]]>=d[b]) a=f[a][i];
	if(a==b) return a;
	for(int i=20;i>=0;i--)
		if(f[a][i]!=f[b][i]) a=f[a][i],b=f[b][i];     //往上找共同根的位置
	return f[a][0]; 
}
int main()
{
	cin>>n>>k>>m>>s;
	for(int i=1;i<=k;i++)
	{
		int x,y;
		cin>>x>>y;
		a[x].push_back(y);
		a[y].push_back(x);
	}
	dfs(s);
	while(m--)
	{
		int a,b;
		cin>>a>>b;
		if(a==s||b==s) cout<<s<<endl;
		else if(a==b) cout<<a<<endl;
		else cout<<lca(a,b)<<endl;
	}
	return 0;
}