70分求助,照着进阶指南敲的
查看原帖
70分求助,照着进阶指南敲的
439000
User439000楼主2023/8/23 14:58
#include<bits/stdc++.h>
using namespace std;
int d[600000],dis[600000],n,s,m,ver[110000],head[1100000],nex[1100000],tot,f[500005][30],t;
void add(int a,int b){
	ver[++tot]=b;
	nex[tot]=head[a];
	head[a]=tot;
}
void bfs(int s){
	queue<int>q;
	q.push(s);
	d[s]=1;
	while(!q.empty()){
		int x=q.front();
		q.pop();
		for(int i=head[x];i;i=nex[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(y==x)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(){
	cin>>n>>m>>s;
	
	t=(int)(log(n)/log(2))+1;
	//for(int i=0;i<=t;i++)f[s][i]=s;
	int a,b;
	for(int i=1;i<=n-1;i++)
	{
		cin>>a>>b;
		
		add(a,b);
		add(b,a);
	}
	bfs(s);
	for(int i=1;i<=m;i++){
		cin>>a>>b;

		cout<<lca(a,b)<<endl;
	}
}
2023/8/23 14:58
加载中...