刚学LCA一秒的蒟蒻求助!!
查看原帖
刚学LCA一秒的蒟蒻求助!!
181037
ysmlwudia楼主2020/10/17 10:33

样例没过,但我查了好几遍不知道哪里错了,倍增求法

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int n,m,s,dis[500005],last[500005],tot,f[500005][20];
struct edge{
	int prev,to;
}e[1000005];
void add(int a,int b){
	e[++tot]=edge{last[a],b};
	last[a]=tot;
}
void init(int u,int fa){
	dis[u]=dis[fa]+1;
	for(int i=1;i<=20;i++){
		f[u][i]=f[f[u][i-1]][i-1];
	}
	for(int i=last[u];i;i=e[i].prev){
		int v=e[i].to;
		if(v==fa)continue;
		f[v][0]=u;
		init(v,u);
	}
}
int lca(int x,int y){
	if(dis[x]<dis[y])swap(x,y);
	for(int i=20;i>=0;i--){
		if(dis[f[x][i]]>=dis[y]){
			x=f[x][i];
		}
		if(x==y) return x;
	}
	for(int i=20;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;
	for(int i=1,x,y;i<n;i++){
		cin>>x>>y;
		add(x,y);
		add(y,x);
	}
	init(s,0);
	for(int i=1,a,b;i<=m;i++){
		cin>>a>>b;
		cout<<lca(a,b)<<endl;
	}
	return 0;
}

2020/10/17 10:33
加载中...