LCA(最近公共祖先) 模板

6 篇文章 0 订阅
本文深入介绍了LCA(最近公共祖先)算法的三种实现方法:倍增法、RMQ(区间最小值查询)和树链剖分。每种方法都详细阐述了其核心思想、时间复杂度,并提供了完整的C++代码实现。适合对LCA算法感兴趣的读者,以及需要解决树状结构中查询问题的开发者。
摘要由CSDN通过智能技术生成

对LCA及算法的介绍:

https://www.cnblogs.com/zhouzhendong/p/7256007.html

https://www.cnblogs.com/scau20110726/archive/2013/05/26/3100812.html

模板:

倍增:

洛谷 P3379

核心思想:二进制,预先处理出树上每个节点的往上走1、 2、 4、……2^n层的点,就可得到所有结果并提高效率。

时间复杂度:预处理O(nlgn)   每次查询(lgn)

#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;

const int maxn=5e5+5;

struct edge
{
    int to,nxt;
}Edge[maxn<<1];

int n,m,root,tot=0;
int head[maxn];
int deep[maxn];
int fa[maxn][30];
int bs[30];

inline void addedge(int u,int v)
{
    Edge[++tot].to=v,Edge[tot].nxt=head[u],head[u]=tot;
    Edge[++tot].to=u,Edge[tot].nxt=head[v],head[v]=tot;
}

void dfs(int cur,int father)
{
    deep[cur]=deep[father]+1;
    fa[cur][0]=father;
    for(int i=1;i<=20;i++)
	fa[cur][i]=fa[fa[cur][i-1]][i-1];
    for(int i=head[cur];i;i=Edge[i].nxt)
        if(Edge[i].to!=father)
            dfs(Edge[i].to,cur);
}
inline int skip(int x,int level)
{
    for(int i=20;i>=0;i--)
    {
        if(bs[i]&level)
            x=fa[x][i];
    }
    return x;
}

inline int LCA(int u,int v)
{
    if(deep[u]<deep[v])
        swap(u,v);
    u=skip(u,deep[u]-deep[v]);
    if(u==v)
        return u;
    for(int i=20;i>=0;i--)
        if(fa[u][i]!=fa[v][i])
            u=fa[u][i],v=fa[v][i];
    return fa[u][0];
}

int main()
{
    for(int i=0;i<=20;i++)
        bs[i]=1<<i;
    scanf("%d%d%d",&n,&m,&root);
    int u,v;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        addedge(u,v);
    }
    dfs(root,0);
    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&u,&v);
        printf("%d\n",LCA(u,v));
    }
    return 0;
}

RMQ:

例题同上

时间复杂度:预处理O(nlgn)     查询O(1) 当查询远大于顶点个数时再考虑此种方法。

#include<iostream>
#include<cstdio>
#include<vector>
#include<cmath>
using namespace std;

const int maxn=5e5+5;

struct edge
{
    int to,nxt;
}Edge[maxn<<1];

int n,m,root,tot=0,cnt=0;
int head[maxn];
int deep[maxn];
int first[maxn];
int oula[maxn<<1];
int st[maxn<<1][30];
int bs[30];

inline int read()
{
    int tmp=0,f=1;
    char ch=getchar();
    while(!isdigit(ch))
    {
        if(ch=='-')
            f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
        tmp=tmp*10+ch-'0',ch=getchar();
    return f*tmp;
}
inline void addedge(int u,int v)
{
    Edge[++tot].to=v,Edge[tot].nxt=head[u],head[u]=tot;
    Edge[++tot].to=u,Edge[tot].nxt=head[v],head[v]=tot;
}

void dfs(int cur,int father)
{
    oula[++cnt]=cur;
    first[cur]=cnt;
    deep[cur]=deep[father]+1;
    for(int i=head[cur];i;i=Edge[i].nxt)
    {
        if(Edge[i].to!=father)
        {
            dfs(Edge[i].to,cur);
            oula[++cnt]=cur;
        }
    }
}

inline void initst()
{
    int tmp1,tmp2;
    for(int i=1;i<=cnt;i++)
        st[i][0]=oula[i];
    for(int j=1;j<=20;j++)
    {
        for(int i=1;i+bs[j]-1<=cnt;i++)
        {
            tmp1=st[i][j-1];
            tmp2=st[i+bs[j-1]][j-1];
            if(deep[tmp1]<deep[tmp2])
                st[i][j]=tmp1;
            else
                st[i][j]=tmp2;
        }
    }
}

inline int LCA(int u,int v)
{
    int l=first[u];
    int r=first[v];
    if(l>r)
        swap(l,r);
    int MAX=log2(r-l+1);
    int tmp1=st[l][MAX];
    int tmp2=st[r-bs[MAX]+1][MAX];
    if(deep[tmp1]<deep[tmp2])
        return tmp1;
    else
        return tmp2;
}

int main()
{
    for(int i=0;i<=20;i++)
        bs[i]=1<<i;
    scanf("%d%d%d",&n,&m,&root);
    int u,v;
    for(int i=1;i<n;i++)
    {
        u=read();
        v=read();
        addedge(u,v);
    }
    dfs(root,0);
    initst();
    for(int i=0;i<m;i++)
    {
        u=read();
        v=read();
        printf("%d\n",LCA(u,v));
    }
    return 0;
}

树链剖分:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;

const int maxn=5e5+5;

struct edge
{
    int to,nxt;
}Edge[maxn<<1];

int n,m,tot=0,cnt=0;
int head[maxn];
int siz[maxn];//子树大小
int son[maxn];//重儿子
int fa[maxn];//父节点
int deep[maxn];//深度
int top[maxn];//所在链链顶
int pos[maxn];//dfs序编号

inline void addedge(int u,int v)
{
    Edge[++cnt].to=v,Edge[cnt].nxt=head[u],head[u]=cnt;
    Edge[++cnt].to=u,Edge[cnt].nxt=head[v],head[v]=cnt;
}

void dfs1(int u,int f)
{
    siz[u]=1,son[u]=0,deep[u]=deep[f]+1,fa[u]=f;
    for(int i=head[u];i;i=Edge[i].nxt)
    {
        if(Edge[i].to!=f)
        {
            dfs1(Edge[i].to,u);
            siz[u]+=siz[Edge[i].to];
            if(siz[Edge[i].to]>siz[son[u]])
                son[u]=Edge[i].to;
        }
    }
}

void dfs2(int u,int f,int k)
{
    pos[u]=++tot;
    top[u]=k;
    if(son[u])
        dfs2(son[u],u,k);
    for(int i=head[u];i;i=Edge[i].nxt)
        if(Edge[i].to!=f&&Edge[i].to!=son[u])
            dfs2(Edge[i].to,u,Edge[i].to);
}

inline int lca(int u,int v)
{
    int f1=top[u],f2=top[v];
    while(f1!=f2)
    {
        if(deep[f1]<deep[f2])
            swap(f1,f2),swap(u,v);
        u=fa[f1];
        f1=top[u];
    }
    if(deep[u]<deep[v])
        return u;
    else
        return v;
}

int main()
{
    int u,v,root;
    scanf("%d%d%d",&n,&m,&root);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        addedge(u,v);
    }
    dfs1(root,0);
    dfs2(root,0,root);
    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&u,&v);
        printf("%d\n",lca(u,v));
    }
    return 0;
}

Tarjan离线:

待补。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值