最近公共祖先（LCA）

最新推荐文章于 2024-09-07 10:59:43 发布

dadidididi_kxj

最新推荐文章于 2024-09-07 10:59:43 发布

阅读量50

点赞数

分类专栏：数据结构文章标签： c++

本文链接：https://blog.csdn.net/m0_65170489/article/details/126425162

版权

数据结构专栏收录该内容

5 篇文章 0 订阅

订阅专栏

倍增算法

可以强制在线，也可以不强制在线。

#include<bits/stdc++.h>
using namespace std;
#define N 500001
int n,m,tot;
int h[N],nt[N<<1],to[N<<1];
void add(int u,int v)
{
	nt[++tot]=h[u];
	h[u]=tot;
	to[tot]=v;
}
int d[N],f[N][23];
void dfs(int x,int fa)
{
	d[x]=d[fa]+1;
	f[x][0]=fa;
	for(int i=1; i<=22; i++)
	{
		f[x][i]=f[f[x][i-1]][i-1];
		if(f[x][i]==0)break;
	}
	for(int i=h[x],y; i; i=nt[i])
	{
		y=to[i];
		if(y==fa)continue;
		dfs(y,x);
	}
}
int LCA(int x,int y)
{
	if(d[x]<d[y])swap(x,y);
	for(int i=22; i>=0; i--)
	if(d[f[x][i]]>=d[y])x=f[x][i];
	if(x==y)return x;
	for(int i=22; i>=0; i--)
	{
		if(f[x][i]==f[y][i])continue;
		x=f[x][i],y=f[y][i];
	}
	return f[x][0];
}
int root;
int main()
{
	scanf("%d%d%d",&n,&m,&root);
	for(int i=1,x,y; i<n; i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y);add(y,x);
	}
	dfs(root,0);
	for(int j=1; j<=n; j++){for(int i=0; i<=11; i++)cout<<f[j][i]<<" ";cout<<endl;}
	for(int i=1,x,y; i<=m; i++)
	{
		scanf("%d%d",&x,&y);
		printf("%d\n",LCA(x,y));
	}
	return 0;
}

离线算法 – $T a r jan$ 算法

对于节点 $x$ , $x, y$ （ $y$ 已经遍历过）的最近公共祖先一定是 $f in d (y)$ 。

原因如下：

对于正在判断 $L C A$ 的节点 $x$ 。

$x$ 没有回溯，所以 $f a [x] = x$ 。

对于以 $x$ 为根的子树的任意一个节点 $u$ ，有 $f in d (u) = x$ ,即 $u$ 的祖先中还未回溯的节点是 $x$ 。

当 $y$ 是以 $x$ 为根的子树的节点时， $f in d (y)$ 指向 $x$ ， $x, y$ 的 $L C A$ 是 $x$ 。
当 $y$ 是 $x$ 的祖先时， $f a [y] = y$ ,因为 $y$ 节点还未回溯， $x, y$ 的 $L C A$ 是 $y$ 。
当 $y$ 既不是是以 $x$ 为根的子树的节点，也不是以 $x$ 的祖先时，因为 $f in d (y)$ 是指 $y$ 的祖先中还未回溯的节点，该节点一定是从根节点到 $x$ 路径中的节点（该路径中的节点还未回溯），该节点一定是 $x, y$ 的共同祖先，并且 $f in d (y)$ 指向的是离 $y$ 最近的节点 $L C A (x, y)$ ，一定不是 $L C A (x, y)$ 的祖先（因为 $L C A (x, y)$ 比 $L C A (x, y)$ 的祖先离 $y$ 更近），所以 $L C A (x, y) = f in d (y)$ 。

#include<bits/stdc++.h>
using namespace std;
#define N 500001
vector < pair <int , int > > e[N];
int tot,h[N],to[N*2],nt[N*2];
int n,fa[N];
void add(int u,int v)
{
	nt[++tot]=h[u];
	h[u]=tot;
	to[tot]=v;
}
void add_e(int x,int y,int i)
{
	e[x].push_back(make_pair(y,i));
	e[y].push_back(make_pair(x,i));
}
int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
int d[N],v[N],len[N],lca[N*4];
void Tarjan(int x,int p)
{
	d[x]=d[p]+1;
	v[x]=1;
	for(int i=h[x],y; i; i=nt[i])
	{
		y=to[i];
		if(v[y])continue;
		Tarjan(y,x);
		fa[y]=x;
	}
	for(int i=0; i<e[x].size(); i++)
	{
		int y=e[x][i].first;
		if(v[y])
		{
			lca[e[x][i].second]=find(y);
		}
	}
}
int m,root;
int main()
{
	cin>>n>>m>>root;
	for(int i=1; i<=n; i++)fa[i]=i;
	for(int i=1,x,y; i<n; i++)
	{
		cin>>x>>y;
		add(x,y);add(y,x);
	}
	for(int i=1,x,y; i<=m; i++)
	{
		cin>>x>>y;
		add_e(x,y,i);
	}
	Tarjan(root,0);
	for(int i=1; i<=m; i++)
	{
		cout<<lca[i]<<endl;
	}
}