LCA倍增

LCA倍增模板题

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
const ll maxn=5e5+10;
ll n,m,s;
ll cnt=0;
ll head[maxn],fa[maxn][23],lg2[maxn],vis[maxn],depth[maxn];
struct edge{
	ll v,nex;
}e[maxn<<1];
void add(ll u,ll v){
	cnt++;
	e[cnt].v=v;
	e[cnt].nex=head[u];
	head[u]=cnt;
}
void dfs(ll u){
//	vis[u]=1;
	for(ll i=head[u];i;i=e[i].nex){
		ll v=e[i].v;
		if(vis[v]==1) continue;
		vis[v]=1;
		fa[v][0]=u;
		depth[v]=depth[u]+1;
		for(ll i=1;i<=lg2[depth[v]];i++){
			fa[v][i]=fa[fa[v][i-1]][i-1];
		}
		dfs(v);
	}
}
ll lca(ll u,ll v){
	if(depth[u]<depth[v]){
		ll t=u;
		u=v;
		v=t;
	}
	while(depth[u]!=depth[v]){
		u=fa[u][lg2[depth[u]-depth[v]]];
	}
	if(u==v) return u;
	for(ll i=lg2[depth[u]];i>=0;i--){
		if(fa[u][i]!=fa[v][i]){
			u=fa[u][i];
			v=fa[v][i];
		}
	}
	return fa[u][0];
}
int main()
{
	cin>>n>>m>>s;
	for(ll i=1;i<n;i++){
		ll u,v;
		cin>>u>>v;
		add(u,v);
		add(v,u);
	}	
	for(ll i=2;i<=n;i++){
		lg2[i]=lg2[i/2]+1;
	}
	vis[s]=vis[0]=1;
	fa[s][0]=0;
	depth[s]=1;
	depth[0]=0;
	dfs(s);
	for(ll i=1;i<=m;i++){
		ll u,v;
		cin>>u>>v;
		cout<<lca(u,v)<<endl;
	}
	return 0;
}