只有subtask1A一个点,其余全wa,求调
查看原帖
只有subtask1A一个点,其余全wa,求调
797897
_QyGyQ_楼主2023/9/3 11:57
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
const int N=5e5+7;
struct EDGE{
	int to;
	int next;
}edge[N<<1];
int head[N<<1],cnt;
void init(){
	cnt=0;
	for(int i=0;i<N;i++){
		edge[i].next=-1;
		head[i]=-1;
	}
}
void addedge(int u,int v){
	edge[cnt].to=v;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}
int fa[N][25];
int deep[N];
void dfs(int x,int father){
	deep[x]=deep[father]+1;
	fa[x][0]=father;
	for(int i=1;(1<<i)<=deep[x];i++){
		fa[x][i]=fa[fa[x][i-1]][i-1];
	}
	for(int i=head[x];~i;i=edge[i].next){
		if(edge[i].to!=father){
			dfs(edge[i].to,x);
		}
	}
}
int LCA(int x,int y){
	if(deep[x]<deep[y]) swap(x,y);
	for(int i=19;i>=0;i--){
		if(deep[x]-(1<<i)>=deep[y]){
			x=fa[x][i];
		}
	}
	if(x==y) return x;
	for(int i=19;i<=0;i--){
		if(fa[x][i]!=fa[y][i]){
			x=fa[x][i];
			y=fa[y][i];
		}
	}
	return fa[x][0];
}
int main(){
	init();
	int n,m,root;
	scanf("%d%d%d",&n,&m,&root);
	for(int i=1;i<n;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		addedge(u,v);
		addedge(v,u);
	}
	dfs(root,0);
	while(m--){
		int x,y;
		scanf("%d%d",&x,&y);
		printf("%d\n",LCA(x,y));
	}
	return 0;
}

2023/9/3 11:57
加载中...