洛谷 P3379 最近公共祖先 【LCA】

题目来源:https://www.luogu.org/problem/P3379
洛谷 P3379 最近公共祖先 【LCA】_第1张图片
洛谷 P3379 最近公共祖先 【LCA】_第2张图片
★第一道LCA题,看了半天,这是一道入门滴模板题


LCA有很多解法,本题仅用 倍增法求LCA (我目前只会这一种

思路:

仔细地看样例,可以看明白什么是LCA了吧。

先说说直观的暴力思路:

每个点有一个dep,代表 深度 。现在要求 LCA(a,b ,那么如果 dep(a)dep[a]>=dep[b] )。然后a往上跳(a=a的祖先),一直到 dep[a]=dep[b] 最后一起往上跳,直到他们的祖先相同。

优化方法:

如果我们在跳的过程中可以 一次跳多步 ,那效率肯定会高很多,这就是 树上倍增法
首先 fa[ i ][ j ] 表示 i 的 第 2j 辈的祖先(从i往根节点的方向走 2j 步 到达的点,若不存在则为0),可知fa[ i ][ 0 ]表示父节点。
可以得到递推式 fa[ i ][ j ]=fa[ fa[ i ][ j-1 ] ][ j-1 ] 由此 可以从根节点推出全部的值

预处理部分差不多就是这样~
还想优化的话 可以加一个 lg数组 lg[ i ] 代表 log2(i)+1 递推如右:lg[i]=lg[i-1]+(1<
详见代码
其他方法 如 tarjan rmq 可见大佬博客:https://blog.csdn.net/ywcpig/article/details/52336496


代码:

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
const int maxn=5e5+5;
const int sz=1<<16;
const int inf=2e9;
const int mod=1e9+7;
const double pi=acos(-1);
typedef long long LL;
int n,m,cnt;
struct node
{
    int next;
    int to;
}edge[maxn<<1];
int head[maxn],dep[maxn];
int fa[maxn][22],lg[maxn];
template<class T>
inline void read(T &x)
{
    char c;x=1;
    while((c=getchar())<'0'||c>'9') if(c=='-') x=-1;
    T res=c-'0';
    while((c=getchar())>='0'&&c<='9') res=res*10+c-'0';
    x*=res;
}
void add_edge(int a,int b)              //链式前向星
{
    edge[++cnt].to=b;
    edge[cnt].next=head[a];
    head[a]=cnt;
}
void dfs(int p,int f)                           //预处理
{
//    cout<
    dep[p]=dep[f]+1;
    fa[p][0]=f;
    for(int i=1;(1<<i)<=dep[p];i++){
        fa[p][i]=fa[fa[p][i-1]][i-1];
    }
    for(int i=head[p];i;i=edge[i].next){          
        if(edge[i].to!=f)                         //不能等于父节点
        dfs(edge[i].to,p);
    }
}
int lca(int a,int b)
{
    if(dep[a]<dep[b]) swap(a,b);            //保证 a深度大
    while(dep[a]>dep[b]){                  //a往上跳
        a=fa[a][lg[dep[a]-dep[b]]-1];
    }
    if(a==b) return a;                    
    for(int i=lg[dep[a]]-1;i>=0;i--){     
        if(fa[a][i]!=fa[b][i])
            a=fa[a][i],b=fa[b][i];      //a b 一起往上跳
    }
    return fa[a][0];                 //注意返回的是 a的父节点 
}
int main()
{
    int s,a,b;
    cnt=0;
    read(n); read(m); read(s);
    for(int i=1;i<n;i++){
        read(a); read(b);
        add_edge(a,b);
        add_edge(b,a);
    }
    dfs(s,0);
    for(int i=1;i<=n;i++)
        lg[i]=lg[i-1]+(1<<lg[i-1]==i);
    while(m--){
        read(a); read(b);
        printf("%d\n",lca(a,b));
    }
    return 0;
}

你可能感兴趣的:(=====,数据结构,=====)