倍增法LCA【模板】

在這裏插入圖片描述

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+10;
struct zzz{int t,nex;}e[N<<1];
int head[N],tot;
int depth[N],fa[N][22],lg[N];
inline void read(int &x)
{
	int w=1;x=0;char ch=getchar();
	while(!isdigit(ch))	{if(ch=='-')w=-1;ch=getchar();}
	while(isdigit(ch))	x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
	x*=w;	
}
inline void add(int x,int y)
{
	e[++tot].t=y;e[tot].nex=head[x];head[x]=tot;
}
inline void dfs(int now,int fath)
{
	fa[now][0]=fath;depth[now]=depth[fath]+1;
	for(int i=1;i<=lg[depth[now]];i++)
		fa[now][i]=fa[fa[now][i-1]][i-1];
	for(int i=head[now];i;i=e[i].nex)
		if(e[i].t!=fath)	dfs(e[i].t,now);
}
int LCA(int x,int y)
{
	if(depth[x]<depth[y])	swap(x,y);
	while(depth[x]>depth[y])	
		x=fa[x][lg[depth[x]-depth[y]]-1];
	if(x==y)	return x;
	for(int k=lg[depth[x]]-1;k>=0;k--)
		if(fa[x][k]!=fa[y][k])
			x=fa[x][k],y=fa[y][k];
	return fa[x][0];
}
int main()
{
	int n,m,s;
	read(n); read(m); read(s);
	for(int i=1,x,y;i<=n-1;i++)
	{
		read(x); read(y);
		add(x,y); add(y,x);
	}
	for(int i=1;i<=n;i++)
		lg[i]=lg[i-1]+(1<<lg[i-1]==i);
	dfs(s,0);
	for(int i=1,x,y;i<=m;i++)
	{
		read(x); read(y);
		printf("%d\n",LCA(x,y));
	}
} 
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章