【LCA】最近公共祖先问题Lowest Common Ancestors

5 篇文章 0 订阅
1 篇文章 0 订阅

昨天集训出了一道LCA的题目,我只会写暴力然后不知道为啥崩栈了,然后想起了去年NOIP Day1 T3 也是不会LCA就废了,然后就没有然后了.....于是我今天学习了各种LCA的做法。

LCA:对于有根树T的两个结点u、v,最近公共祖先LCA(T,u,v)表示一个结点x,满足x是u、v的祖先且x的深度尽可能大。——百度百科

举个例子:


3和4节点的LCA就是2,2和6号节点的LCA是1,5和6的LCA就是5.

暴力求LCA的方法:

对树进行遍历求出每个节点的深度,如下图:


求任意两个节点的LCA,先把深度大的沿父亲移到同样的深度,若两个节点不是同一个结点就把两个结点同时向父亲移,最后结点相同得到的就是LCA。

举上图的例子,求3结点和5结点的LCA就,3结点深度大于5,先跳到2结点,两结点不同,继续跳,两节点都跳到1结点,1就是他们的LCA。

Code C++:

void dfs(int p,int dth){//对树进行遍历求深度
	dep[p]=dth;
	for(int i=first[p];i;i=e[i].next){
		if(!dep[e[i].p]){
			father[e[i].p]=p;
			val[e[i].p]=e[i].l;
			dfs(e[i].p,dth+1);
		}
	}
}
int lca(int x,int y){
	while(dep[x]>dep[y])x=father[x];
	while(dep[x]<dep[y])y=father[y];
	while(x!=y){x=father[x];y=father[y];}
	return x;
}

暴力 求一次LCA时间复杂度是O(树的深度)的,显然对于大数据会挂,那怎么办呢。

我们有LCA+RMQ(区间最值)的倍增在线算法

记得求区间最值的ST算法吗???

min[i][j]=min(min[i][j-1],min[i+(1<<(j-1))][j-1]);
类似地,我们可以利用倍增的算法求出 i 结点往上的第 2^j 个结点是谁

另fa[i][j]为i结点往上跳2^j的结点编号

易得fa[i][j]=fa[ fa[ i ][ j-1 ] ][ j-1 ](i结点往上跳2^(j-1)下再跳2^(j-1)就是i结点往上跳2^j的结果

我们就能在O(nlogn)的时间预处理出这个东西

然后对每一个LCA的查询,还是先跳到相同深度,不过这次类似于二分查找地在logn的时间跳到相同深度结点,再把两个结点一起跳到使两个结点不同的最后两个结点,最终LCA就是fa[i][0].

时间复杂度:预处理O(nlogn)、一次查询(logn)

具体实现看下面代码。

我们先来看看2013Noip day1 T3 Trunk(Wikioi3282)

题目大意:给一个图,求最大生成树,询问生成树上两个结点路径上最小的边权,若两点不可达输出-1。

题解:先求最大生成树,然后对生成树dfs,并进行倍增同时记录g[i][j]表示以i为结点到i往上2^j个结点之间的最小边权,直接回答询问即可。

C++代码:

3287 货车运输NKWBTB 测试通过 Accepted
100
» 561ms 4296kb C++ 2068B  

#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#define MAXM 50001
#define MAXN 10001
#define DA 17
//DA开到2^n比N大就行 
using namespace std;
int first[MAXN],father[MAXN],ct=1,fa[MAXN][DA],lth[MAXN][DA],dep[MAXN];
//father为并查集数组,fa是结点父亲倍增数组,lth为最小边权记录数组,dep为深度数组 
struct edge{
	int p,next,l;
	void add(int u,int v,int val){
		p=v;
		next=first[u];
		l=val;
	}
}e[MAXM],line[MAXM*2];
bool cmp(const edge &a,const edge &b){return a.l>b.l;}
int findset(int x){
	if(x!=father[x])father[x]=findset(father[x]);
	return father[x];
}
 
bool Union(int x,int y){
	int fx=findset(x);
	int fy=findset(y);
	if(fx==fy)return false;
	father[fx]=fy;
	return true;
}
void dfs(int p,int dth){
	dep[p]=dth;
	for(int i=first[p];i;i=e[i].next){
		if(!fa[e[i].p][0]){
			fa[e[i].p][0]=p;
			lth[e[i].p][0]=e[i].l;
			dfs(e[i].p,dth+1);
		}
	}
}
//对树进行遍历预处理出深度,初始化边权数组和倍增数组 
int query(int u,int v){
	int len=0x7fffffff;
	if(dep[u]<dep[v])swap(v,u);
	for(int i=DA-1;i>=0;i--)//将两结点跳到同样深度同时更新最小边权大小 
		if(dep[fa[u][i]]>=dep[v]){
			len=min(len,lth[u][i]);
			u=fa[u][i];
		}
	if(u!=v){//两节点同时跳到最后不同的两个结点 
	for(int j=DA-1;j>=0;j--)
		if(fa[u][j]!=fa[v][j]){
			len=min(len,min(lth[u][j],lth[v][j]));
			u=fa[u][j];v=fa[v][j];
		}
	len=min(len,min(lth[u][0],lth[v][0]));
	}
	return len;
}
//查询 
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=0;i<m;i++)scanf("%d%d%d",&line[i].p,&line[i].next,&line[i].l);
	sort(line,line+m,cmp);
	for(int i=1;i<=n;i++)father[i]=i;
	int cnt=0;
	for(int i=0;i<m;i++){
		if(Union(line[i].p,line[i].next)){
			++cnt;
			e[ct].add(line[i].p,line[i].next,line[i].l);
			first[line[i].p]=ct++;
			e[ct].add(line[i].next,line[i].p,line[i].l);
			first[line[i].next]=ct++;
		}
		if(cnt==n-1)break;
	}
	for(int i=1;i<=n;i++)if(!fa[i][0]){lth[i][0]=0x7fffffff;fa[i][0]=i;dfs(i,1);}
	//对树进行预处理 
	for(int i=1;i<DA;i++){
		for(int j=1;j<=n;j++){
			fa[j][i]=fa[fa[j][i-1]][i-1];
			lth[j][i]=min(lth[j][i-1],lth[fa[j][i-1]][i-1]);
		}
	}
	//倍增预处理 
	int q;
	scanf("%d",&q);
	while(q--){
		int u,v;
		scanf("%d%d",&u,&v);
		if(findset(u)!=findset(v)){printf("-1\n");continue;}
		printf("%d\n",query(u,v));
	}
	return 0;
}
//其实这题有更好的做法,详情请移步http://blog.csdn.net/nkwbtb/article/details/21883807


如果查询不要求强制在线的话可以有离线的tarjan算法

时间复杂度为O(n+q)

参考文章:

http://en.wikipedia.org/wiki/Tarjan%27s_off-line_lowest_common_ancestors_algorithm

http://scturtle.is-programmer.com/posts/30055.html

这个算法思想大概是这样的

先把询问都存储下来

在遍历树的时候,维护一个并查集,当遍历原来树的一个子树的时候,把子树中的所有元素并为一个集合,把集合的根指向为子树的根,

那么我可以回答已经遍历过的点与子树跟的LCA询问,若询问点没有遍历过那么先放在一边不管。

例:


箭头表示当前遍历到的结点,相同颜色的点表示已经并入同一集合,如题,现在遍历到点6,回答与结点6有关的询问,假若询问3和6号点的LCA 先检查3号点是否遍历过,然后答案为3所在并查集的祖先,也就是1号点。


又来一题:

WIKIOI 1036

题目大意:给一棵树,给定结点顺序,求从根节点按结点顺序走的最终距离是多少,其中每条边的长度都是1.

解法:显然求相邻点的LCA,然后他们之间的距离就是i到根的距离+j到根的距离-2*LCA到根的距离,累加得到答案即可。

代码:

1036 商务旅行NKWBTB 测试通过 Accepted
100
» 55ms 3008kb C++ 1252B

#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <vector>
#define MAXN 30001
using namespace std;
int first[MAXN],fa[MAXN],ct=1,anc[MAXN],lth[MAXN],ans=0;
bool vis[MAXN];
struct edge{
	int p,next;
	void add(int u,int v){
		p=v;
		next=first[u];
	}
}e[MAXN*2];
vector<int>vec[MAXN];
vector<int>::iterator it;
int findset(int x){
	if(x!=fa[x])fa[x]=findset(fa[x]);
	return fa[x];
}
bool Union(int x,int y){
	int fx=findset(x);
	int fy=findset(y);
	if(fx==fy)return false;
	fa[fx]=fy;
	return true;
}
void tarjan(int p,int f,int l){
	fa[p]=p;lth[p]=l;//处理距离 
	for(int i=first[p];i;i=e[i].next){
		if(!fa[e[i].p]&&e[i].p!=f){
			tarjan(e[i].p,p,l+1);
			Union(p,e[i].p);//合并子树结点到并查集中 
			anc[findset(p)]=p;//并查集根指向子树根节点 
		}
	}
	vis[p]=true;
	for(it=vec[p].begin();it!=vec[p].end();it++){//回答与p相关询问 
		if(vis[*it])ans+=lth[p]+lth[*it]-2*lth[anc[findset(*it)]];
	}
}
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		e[ct].add(u,v);
		first[u]=ct++;
		e[ct].add(v,u);
		first[v]=ct++;
	}
	int s=1,t,q;
	scanf("%d",&q);
	while(q--){//存储询问 
		scanf("%d",&t);
		vec[s].push_back(t);
		vec[t].push_back(s);
		s=t;
	}
	tarjan(1,1,0); 
	printf("%d",ans);
	return 0;
}

————————————————————————————————————————————————

P.S.:以前XTC大神讲过noip 2013 day1 t3 可以tarjan做,可是我想了很久还是不会。。。。。。求大神指点

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值