求调,样例能过
查看原帖
求调,样例能过
1019257
zhaojiawei1234444楼主2024/7/16 07:54

p3379

#include<bits/stdc++.h>
using namespace std;
const int N=500100;
int n,m,rt;
int x,y;
int tot,head[N*2],nxt[N*2],to[N*2];
int lg[N*2];
int deep[N*2];
void add(int x,int y){
	to[tot]=y;
	nxt[tot]=head[x];
	head[x]=tot++;
}
int fa[N][30];
void dfs(int x,int f){
	fa[x][0]=f;
	deep[x]=deep[f]+1;
	for(int i=1;i<=lg[deep[x]];i++){
		fa[x][i]=fa[fa[x][i-1]][i-1];
	}
	for(int i=head[x];i;i=nxt[i]){
		if(to[i]!=f) dfs(to[i],x);
	}
}
int get_lca(int x,int y){
	if(deep[x]<deep[y]){
		swap(x,y);
	}
	for(int i=20;i>=0;i--){
		if(deep[x]-(1<<i)>=deep[y]){
			x=fa[x][i];
		}
	}
	if(x==y){
		return x;
	}
	for(int i=20;i>=0;i--){
		if(fa[x][i]!=fa[y][i]){
			x=fa[x][i];
			y=fa[y][i];
		}
	}
	return fa[x][0];
}
signed main(){
	ios::sync_with_stdio;
	cin.tie(0);cout.tie(0);
	cin>>n>>m>>rt;
	for(int i=1;i<=n;++i)
		lg[i]=lg[i-1]+(1<<lg[i-1]==i);
	for(int i=1;i<n;i++){
		cin>>x>>y;
		add(x,y);
		add(y,x);
	}
	dfs(rt,0);
	for(int j=1;j<=20;j++){
		for(int i=1;i<=n;i++){
			fa[x][i]=fa[fa[x][x-1]][j-1];
		}
	}
	for(int i=1;i<=m;i++){
		cin>>x>>y;
		cout<<get_lca(x,y)<<"\n\n\n";
	}
	return 0;
}
2024/7/16 07:54
加载中...