倍增全T,求调
查看原帖
倍增全T,求调
1054957
Wrx1011楼主2025/7/28 19:38

RT

#include <bits/stdc++.h>
using namespace std;
#define ll long long
ll f[500050][22],d[500050],lg[500050],head[500050],cnt=1;
struct edge{
	ll v,n;
}e[1000010];
void add(ll u,ll v){
	e[cnt].v=v;
	e[cnt].n=head[u];
	head[u]=cnt++; 
}
ll dfs(ll u,ll fa){
	f[u][0]=fa;
	d[u]=d[fa]+1;
	for(ll i=1;i<=20;i++){
		f[u][i]=f[f[u][i-1]][i-1];
	}
	for(ll i=head[u];i;i=e[i].n){
		ll v=e[i].v;
		if(v!=fa){
			dfs(v,u);
		} 
	} 
}
ll lca(ll u,ll v){
	if(d[u]<d[v]){
		swap(u,v);
	}
	for(ll i=20;i>=0;i--){
		if(d[f[u][i]]>=d[v]){
			u=f[u][i];
		}
	}
	if(u==v){
		return u;
	}
	for(ll i=20;i>=0;i--){
		if(f[u][i]!=f[v][i]){
			u=f[u][i];
			v=f[v][i];
		}
	}
	return f[u][0]; 
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	ll s,z,h;
	cin>>s>>z>>h;
	for(ll i=2;i<=s;i++){
		lg[i]=lg[i/2]+1;
	}
	for(ll i=1;i<s;i++){
		ll u,v;
		cin>>u>>v;
		add(u,v);
		add(v,u); 
	}
	dfs(h,0);
	for(ll i=0;i<z;i++){
		ll u,v;
		cin>>u>>v;
		cout<<lca(u,v)<<"\n";
	}
	return 0;
}
2025/7/28 19:38
加载中...