lca板子 带常数优化

#include<bits/stdc++.h>
using namespace std;
const int N = 2000;
int h[N],nex[N<<1],to[N<<1],tot;
void add(int x,int y){
	to[++tot]=y;nex[tot]=h[x];h[x]=tot;
}
int fa[N][30],dep[N],lg[N];
void dfs(int u,int fath){
	fa[u][0]=fath;dep[u]=dep[fath]+1;
	for(int i = 1; i <= lg[dep[u]]; i++) fa[u][i]=fa[fa[u][i-1]][i-1];
	for(int j = h[u]; j; j=nex[j]) if(to[j]!=fath) dfs(to[j],u);
}
int lca(int x,int y){
	if(dep[x]<dep[y]) swap(x,y);
	while(dep[x]>dep[y])  x=fa[x][lg[dep[x]-dep[y]]-1];
	if(x==y) return x;
	for(int i = lg[dep[x]-1]; i >= 0; i--)
		if(fa[x][i]!=fa[y][i])
		x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
int main(){
	int n;
	scanf("%d",&n);
	for(int i = 1; i < n; i++){
		int u,v;
		scanf("%d%d",&u,&v);
		add(u,v);
		add(v,u);
	}	
	for(int i = 1; i <= n; i++) lg[i]=lg[i-1]+(1<<lg[i-1]==i);//预处理lg可以快不少
	dfs(1,0);
	for(int i = 1; i <= m; i++){
		int u,v;
		scanf("%d%d",&u,&v);
		printf("%d\n",lca(u,v));
	} 
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43824564/article/details/106319374
lca