LCA算法(最近公共祖先)

注:横坐标为 i 值,纵坐标为 x 的值 ,ST表里面的值表示 f[x][i]

一开始U和V分别是18和19,在第11层;LCA(在第3层)的下一层是3后面的4和5,在第4层

执行while语句:

如果x和y的数的深度相等,则执行for循环里面的内容,执行到LCA的下一层,然后在执行return  f[x][0],f[x][0]相当于在走一步,也就到达了他们的LCA处!

题目:

代码:

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
using namespace std;

vector<int>a[500010];
int n,m,s,dep[500010],f[500010][21];
void dfs(int x,int father)
{
	dep[x]=dep[father]+1;
	f[x][0]=father;
	for(int i=1;i<=20;i++)
	f[x][i]=f[f[x][i-1]][i-1];
	int l=a[x].size();
	
	for(int i=0;i<l;i++){
		int v=a[x][i];
		if(v==father) continue;
		dfs(v,x);
	}	
}

int LCA(int x,int y)
{
	if(dep[x]<dep[y]) swap(x,y);
	
	while(dep[x]>dep[y])
	{
		x=f[x][(int)log2(dep[x]-dep[y])];
	}
	
	if(x==y) return x;
	
	for(int i=log2(dep[x]);i>=0;i--)
	{
		if(f[x][i]!=f[y][i])
		{
			x=f[x][i];
			x=f[y][i];
		}
	}
    return f[x][0];
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin>>n>>m>>s;
	for(int i=1;i<n;i++)
	{
		int x,y;
		cin>>x>>y;
		a[x].push_back(y);
		a[y].push_back(x);
	}
	dfs(s,0);
	
	for(int i=1;i<=m;i++)
	{
		int x,y;
		cin>>x>>y;
		cout<<LCA(x,y)<<endl;
	}
		
	return 0;
}

另外一个版本:

注:为什么i从19循环?

答:因为N代表最大层数,为500010。而且2的19次方为50000多一点,这确值为524288,所以开到19就可以了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值