本地测样例输不进去,放洛谷里AC了???倍增
查看原帖
本地测样例输不进去,放洛谷里AC了???倍增
1200401
wnqnld_llx楼主2024/4/12 19:59

倍增

有没有大佬解释一下

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=5e5+10;
int dep[N],fa[N][20];
vector<int> e[N];
void dfs(int u,int f){
	dep[u]=dep[f]+1;
	fa[u][0]=f;
	for(int i=1;i<=19;i++){
		fa[u][i]=fa[fa[u][i-1]][i-1];
	}
	for(int v:e[u]){
		if(v!=f) dfs(v,u);
	}
}
int lca(int u,int v){
	if(dep[u]<dep[v]) swap(u,v);
	for(int i=19;i>=0;i--){
		if(dep[fa[u][i]]>=dep[v]){
			u=fa[u][i];
		}	
	}
	if(u==v) return v;
	for(int i=19;i>=0;i--){
		if(fa[u][i]!=fa[v][i]){
			u=fa[u][i];
			v=fa[v][i];
		}
	}
	return fa[u][0];
}
signed main(){
	int n,m,s,x,y;
	scanf("%d%d%d",&n,&m,&s);
	for(int i=0;i<n-1;i++){
		scanf("%d%d",&x,&y);
		e[x].push_back(y);
		e[y].push_back(x);
	} 
	dfs(s,0);
	int a,b;
	for(int i=1;i<=m;i++){
		scanf("%d%d",&a,&b);
		printf("%d\n",lca(a,b));
	}
	return 0;
}
2024/4/12 19:59
加载中...