LCA问题:就是在一棵树中求两个节点的最近公共祖先(如果其中一个节点是另一个节点的祖先,那他们两个节点的最近公共祖先就是第一个节点)
以如下这棵树为例:求1和14节点的最近公共祖先
首先,关于这棵树的存储结构,我们可以采用邻接表或链式前向星来存储,这里我们采用链式前向星的方式来举例
链式前向星板子:具体关于图的存储结构可以看图的存储结构
接下来是关于怎样求两个节点的最近祖先
我们最直接的想法就是先将深度较大的节点调整到与深度较小的节点同一深度,然后将两个节点同时从底到高依次枚举每个节点直到两个节点相遇,那这样的时间复杂度为O(n)。那有没有一个求解方法可以更快的让两个节点处于同一深度呢?
首先思考如下情形,如果要从0跳到7,我们可以怎么跳,
1.我们可以逐步跳,此时的时间复杂度为O(n);
2.其实我们可以不用一次只跳一步,我们可以以2^n的步数来跳,比如从0开始先跳2^2个格子到4,然后再跳2^1个格子到6,最后跳2^0个格子到7,这样跳的时间复杂度就是logn。用同样的方法,我们就可以更快的使两个节点到达相同的深度,所以我们要先预处理。
f [i] [j]表示从i开始,跳2^j步到达的节点,因为跳2^j=2^(j-1)*2,所以可以先跳到f [i] [j-1],然后再从 f [i] [j-1]跳2^j-1步就到f [i] [j]了,所以f [i] [j]=f [f [i] [j-1]] [j-1] 。用这个递推方便一会求f数组
dfs是用来求 f 数组并求dep(记录每个节点的深度)数组的。v是儿子节点,father是父亲节点,在main函数中调用 dfs(root,0)。
void dfs(int v,int father){
dep[v]=dep[father]+1;//记录儿子节点的深度
for(int i=1;(1<<i)<=dep[v];i++){//求f数组
f[v][i]=f[f[v][i-1]][i-1];//因为递归是从树顶到叶子去遍历的,所以这个递推式可以求出f数组
}
for(int i=head[v];i;i=e[i].next){//i位边的编号,e是边节点
int p1=e[i].to;//p1是与v相连的边的另一个节点
if(p1==father) continue;//上述edge_add采用了双向的连接,所以儿子节点也连上了父节点,
//当遇上父节点时跳过
f[p1][0]=v;//p1向上走一步就是他的父节点v
dfs(p1,v);
}
}
此时我们已经处理完f数组了,接下来就要移动14节点与1节点的深度相同
int up(int x,int d){//x为节点,d为要走的步数
int ret=x;
for(int i=0;(1<<i)<=n;i++){//n为d
if((1<<i)&d!=0) ret=f[ret][i];//枚举可以移动的步数
return ret;
}
最后就可以求两个节点的公共祖先了,两个节点同时向上跳,当跳到的节点不同时,才能继续跳,因为如果跳到的节点相同,可能是跳到了更远的祖先上去了,所以当不同时才可以跳,直到跳到他们最近公共祖先的孩子节点上,最后在向上跳一步就是他们的最近公共祖先
其实将两个节点移动到同深度的过程可以和移动到最近公共祖先的过程结合,代码如下: