浅谈RMQ实现LCA | 易学教程

求两个点的LCA一共有四种方法

Tarjan，倍增，RMQ还有树链剖分（我也不会）

今天我们来学习如何用RMQ来实现LCA

首先我们要知道什么是RMQ（区间最值）

推荐一篇大佬的博客:https://www.cnblogs.com/YSFAC/p/7189571.html

好了，现在大家都知道了什么是RMQ，那么我们来看一下什么是LCA

LCA是树上两个点的最近公共祖先

举个例子：

在这个图中，4和5的最近公共祖先就是2

5和6的最近公共祖先就是1

然后我们来看如何快速的求出两个点的最近公共祖先

首先是求出每个点的深度

这个操作用一个DFS维护即可，每个店的深度是其父亲的深度加1

void dfs(int x,int fa){
    for(int e=head[x];e;e=nxt[e]){
        d[to[e]]=d[x]+1;
        dfs(to[e],x);
    }
    return ;
}

然后是维护f数组

f数组储存的是每个点向上跳2^j步所到达的点，并且我们可以确定这个点是唯一的，因为是树

然后来看如何维护

我们可以用递推的方法来求，时间复杂度是(nlogn)的

对于f[i][j]

是从x往上跳2^j步

可以推导出相当于是从x往上跳2^j-1步，然后再跳2^j-1步

也就是相当于从x往上跳2^j-1步之后，从那个点在往上跳2^j-1步

由此得出了我们的递推式:

f[i][j]=f[f[i][j-1][j-1]

实现起来也就很简单了，f[i][0]已经在DFS里处理过了

void pre(){for(int j=1;j<=17;j++) for(int i=1;i<=n;i++) f[i][j]=f[f[i][j-1]][j-1];}

最后就到了最重要的一步

求两个点的LCA

时间复杂度大约为logn，其实也可以当做O(1)

step1：

将两个点跳到同一高度

为了代码简洁

我们确保x的深度较深

往上跳的条件是跳完之后的深度小于等于y的深度

（因为要确保跳完之后深度一样）

于是代码也就好写了

step2：
两个点一起往上跳

我们的目标是将两个点跳到他们最近公共祖先的两个子节点处

为啥不直接跳上去呢

因为我们的判断条件是两个点跳完之后不相同才可以跳，否则会直接跳到根节点

现在我们的代码已经呼之欲出了

下面给出代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<string>
using namespace std;
inline int rd(){
    int x=0,f=1;
    char ch=getchar();
    for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-1;
    for(;isdigit(ch);ch=getchar()) x=x*10+ch-'0';
    return x*f;
}
inline void write(int x){
    if(x<0) putchar('-'),x=-x;
    if(x>9) write(x/10);
    putchar(x%10+'0');
    return ;
}
int n,m,root;
int total=0,head[1000006],to[1000006];
int nxt[1000006];
int f[1000006][20];
int dep[1000006];
inline void add(int x,int y){
    total++;
    to[total]=y;
    nxt[total]=head[x];
    head[x]=total;
    return ;
}
inline void dfs(int x,int fa){
    f[x][0]=fa;
    for(int e=head[x];e;e=nxt[e]){
        if(to[e]==fa) continue;
        dep[to[e]]=dep[x]+1;
        dfs(to[e],x);
    }
    return ;
}
inline void pre(){for(int i=1;i<=18;i++) for(int j=1;j<=n;j++) f[j][i]=f[f[j][i-1]][i-1];}//预处理出每个点往上跳2^j步会到哪里 
inline int lca(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=17;i>=0;i--) if(dep[f[x][i]]>=dep[y]) x=f[x][i];
    if(x==y) return x;
    for(int i=17;i>=0;i--) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];//因为是到了最近公共祖先的两个子节点，所以还要再往上跳一步 
}
int main(){
    n=rd(),m=rd(),root=rd();
    for(int i=1;i<n;i++){
        int x=rd(),y=rd();
        add(x,y),add(y,x);
    }
    dep[root]=1;
    dfs(root,0);
    pre();
    for(int i=1;i<=m;i++){
        int x=rd(),y=rd();
        write(lca(x,y)),puts("");
    }
    return 0;
}

来源：https://www.cnblogs.com/WWHHTT/p/9804966.html

标签

rmq

lca