lca

倍增法

#include<bits/stdc++.h>
using namespace std;
const int maxn=1000010;
int n,m,s;
int h[maxn],nxt[maxn],to[maxn],tot;
int Log[maxn];
int dep[maxn];
int dis[maxn];
int fa[maxn][20];
void add(int x,int y)
{
	tot++;
	to[tot]=y;
	nxt[tot]=h[x];
	h[x]=tot;
}
void llog(){for (int i=2;i<=n;i++) Log[i]=Log[i>>1]+1;}
void dfs(int now,int f)
{
	dep[now]=dep[f]+1;
	fa[now][0]=f;
	for (int i=1;i<=19;i++)
	{
		fa[now][i]=fa[fa[now][i-1]][i-1];
	}
	for (int i=h[now];i;i=nxt[i])
	{
		int y=to[i];
		if (y==f) continue;
		dis[y]=dis[now]+1;
		dfs(y,now);
	}
}
int lca(int x,int y)
{
	if (dep[x]>dep[y]) swap(x,y);
	int d=dep[y]-dep[x];
	for (int i=Log[d];i>=0;i--)
	{
		if (d>>i&1) y=fa[y][i]; 
	}
	if (x == y) return x;
	for (int i=Log[dep[x]];i>=0;i--)
	{
		if (fa[x][i]!=fa[y][i])
		{
			x=fa[x][i];
			y=fa[y][i];
		}
	}
	return fa[x][0];
}
int main()
{
	cin >> n >> m >> s;
	llog();
	for (int i=1;i<=n-1;i++)
	{
		int x,y;
		cin >> x >> y;
		add(x,y);
		add(y,x);
	}
	dfs(s,s);
	while (m--)
	{
		int x,y;
		cin >> x >> y;
//		int d=dis[x]+dis[y]-dis[lca(x,y)]*2;
//		cout << d << endl;
		cout << lca(x,y) << endl;
	}
	return 0;
}
posted @ 2026-03-24 11:49  msjing  阅读(3)  评论(0)    收藏  举报