LCA(倍增)

模板,,

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm> 
using namespace std;
int t;
int n;
int fa[10005][20];
int fst[10005];
int nxt[10005];
int vv[10005];
int deep[10005]; 
int edge;
void mem()
{
	edge=0;
	memset(fa,0,sizeof(fa));
	memset(vv,0,sizeof(vv));
	memset(deep,0,sizeof(deep));
	memset(nxt,0,sizeof(nxt));
	for(int i=1;i<=n;i++)
	fst[i]=-1;
}
void add(int x,int y)
{
	edge++;
	nxt[edge]=fst[x];
	fst[x]=edge;
	vv[edge]=y;
} 
void dfs(int x,int low)
{
	deep[x]=low;
	for(int i=fst[x];i!=-1;i=nxt[i])
	{
		dfs(vv[i],low+1);
	}
} 
int main()
{
	cin>>t;
	int x,y;
	while(t--)
	{
		int u,v;
		cin>>n;
		mem();
		for(int i=1;i<=n-1;i++)
		{
			scanf("%d%d",&x,&y);
			add(x,y);
			fa[y][0]=x;
		}
		for(int i=1;i<=n;i++)
		{
			if(fa[i][0]==0)
			{
				//cout<<i<<endl;
			dfs(i,1);	
			}
		}
		for(int j=1;j<=16;j++)
		for(int i=1;i<=n;i++)
		{
			fa[i][j]=fa[fa[i][j-1]][j-1];
		}
		cin>>u>>v;
		if(deep[u]<deep[v])swap(u,v);
		for(int i=16;i>=0;i--)
		{
			if(deep[fa[u][i]]>=deep[v])u=fa[u][i];
		}
		if(u==v)printf("%d\n",u);
		else
		{
			for(int i=16;i>=0;i--)
			{
				if(fa[u][i]!=fa[v][i])
				{
					u=fa[u][i];
					v=fa[v][i];
				}
			}
			printf("%d\n",fa[u][0]);
		}
	}
	return 0;
}

 

posted @ 2018-07-20 08:59  Richardluan  阅读(2)  评论(0)    收藏  举报