蒟蒻刚学LCA 70分 2T 1WA 求助
查看原帖
蒟蒻刚学LCA 70分 2T 1WA 求助
209825
cd2020楼主2020/7/28 19:59
#include<bits/stdc++.h>
using namespace std;
const int maxn=5e5+5;
int n,m,s,t,head[maxn],next[maxn],ver[maxn],ru[maxn],tot;
int d[maxn],f[maxn][19];
void add(int x,int y)
{
	ver[++tot]=y,next[tot]=head[x],head[x]=tot;
}
void bfs()
{
	queue<int>q;
	q.push(s);d[s]=1;f[s][0]=s;
	while(q.size())
	{
		int x=q.front();q.pop();
		for(int i=head[x];i;i=next[i])
		{
			int y=ver[i];
			if(d[y])continue;
			d[y]=d[x]+1;
			f[y][0]=x;
			for(int j=1;j<=t;j++)
			{
				f[y][j]=f[f[y][j-1]][j-1];
			}
			q.push(y);
		}
	}
}
int lca(int x,int y)
{
	if(d[x]>d[y])swap(x,y);
	for(int i=t;i>=0;i--)
	{
		if(d[f[y][i]]>=d[x])y=f[y][i];
	}
	if(x==y)return x;
	for(int i=t;i>=0;i--)
	{
		if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
	}
	return f[x][0];
}
int main()
{
	ios::sync_with_stdio(0);
	cin>>n>>m>>s;
	t=(int)log(n)/log(2);
	int x,y;
	for(int i=1;i<n;i++)
	{
		cin>>x>>y; 
		add(y,x);
		add(x,y);
	}
	bfs();
	for(int i=1;i<=m;i++)
	{
		cin>>x>>y;
		cout<<lca(x,y)<<endl;
	}
	return 0;
}

2020/7/28 19:59
加载中...