LCA(倍增)

0 篇文章 0 订阅

模板,,

#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;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值