码风超级无敌良好的倍增,求调
查看原帖
码风超级无敌良好的倍增,求调
826079
Autumn_Rain楼主2024/2/27 20:22
#include<bits/stdc++.h>
using namespace std;
const int N=5e5+10;
int n,m,s,x,y,a,b;
int d[N],anc[N][20];
vector<int>G[N];
void dfs(int u,int fa){
	for(int i=0;i<G[u].size();i++){
		int v=G[u][i];
		if(v==fa)continue;
		d[v]=d[u]+1;
		anc[v][0]=u;
		dfs(v,u);
	}
}
void init(){
	for(int j=1;j<=20;j++){
		for(int i=1;i<=n;i++){
			anc[i][j]=anc[anc[i][j-1]][j-1];
		}
	} 
}
int LCA(int u,int v){
	if(d[u]<d[v])swap(u,v);
	for(int i=20;i>=0;i--){
		if(d[anc[u][i]]>=d[v]){
			u=anc[u][i];
		}
	}
	if(u==v)return u;
	for(int i=20;i>=0;i--){
		if(anc[u][i]!=anc[v][i]){
			u=anc[u][i];
			v=anc[v][i];
		}
	}
	return anc[u][0];
}
int main(){
	cin>>n>>m>>s;
	for(int i=1;i<n;i++){
		cin>>x>>y;
		G[x].push_back(y);
		G[y].push_back(x);
	}
	d[s]=1;
	dfs(s,0);
	init();
	while(m--){
		cin>>a>>b;
		cout<<LCA(a,b)<<endl;
	}
	return 0;
}
2024/2/27 20:22
加载中...