码迷,mamicode.com
首页 > 其他好文 > 详细

LCA入门

时间:2020-03-12 15:58:24      阅读:87      评论:0      收藏:0      [点我收藏+]

标签:不用   调用   struct   ref   答案   没有   wap   memset   main   

这篇文章笔者想总结一下LCA的做法

LCA即最近公共祖先它所要求的是树上任意两个结点的公共祖先 下面提供做法

1、暴力法:

即一层一层往上爬,给定两个结点,先将两个结点的深度调为一致后,一起一层层的向上爬上升,这种做法的时间复杂度为较大,很容易被卡掉,这里就不详细说明,只是放出代码

#include<iostream>
#include
<cstdio> #include<algorithm> #include<cstring> #define ll long long #define maxn 500005 using namespace std; int n,m,s,tot; int head[maxn],deep[maxn],f[maxn],vis[maxn]; struct node { int next,to; } map[maxn*2];//因为是无向图,记得开二倍 void add (int from, int to) { map[tot].next = head[from]; map[tot].to = to; head[from] = tot++; }//存图 void build(int root,int depth) { deep[root] = depth; vis[root] = 1; for(int i=head[root] ;i!=-1;i=map[i].next) { if(vis[map[i].to]) continue; f[map[i].to] = root; build(map[i].to,depth+1); } }//建树,也就是一个dfs的过程,要存储深度 int LCA(int s1,int s2) { while(deep[s1]>deep[s2]) s1 = f[s1]; while(deep[s1]<deep[s2]) s2 = f[s2];//先调到同一深度 while(s1!=s2) { s1 = f[s1]; s2 = f[s2]; }//向上爬 return s1; } int main() { cin>>n>>m>>s; int a,b; memset(head,-1,sizeof(head)); for(int i=1; i<=n-1; i++) { cin>>a>>b; add(a,b); add(b,a); } build(s,1); int son1,son2; for(int i=1 ; i<=m; i++) { cin>>son1>>son2; cout<<LCA(son1,son2)<<endl; }return 0; }

2、现在来说一说一种普遍做法,树上倍增法

在看这里之前,请先保证你对ST表或者RMQ问题有过一定的了解,否则直接看树上倍增会有些难

具体思路和写一个ST表的思路是一样的,只不过这里用fa[i][j]表示第i个结点向上2^j的父亲是谁,即fa[i][0]表示i的父亲,fa[i][1]表示i的爷爷以此类推

相比于暴力做法,我们在dfs建树的过程中又多了一步记录每个点向上2^j的父亲因为考虑到这里要用log2进行处理,我们不妨直接先处理lg数组(即log2(n)是多少)

    lg[1]=0;
     for(int i = 2; i <= n; ++i)
        lg[i] = lg[i>>1]+1;

当然为了方便使用我们还有一种处理方法

    for(int i = 1; i <= n; ++i)
        lg[i] = lg[i-1] + (1 << lg[i-1] == i)  

这样处理的结果就是我们可以直接得到log2(x)+1的值

当然你也可以使用cmath里的log函数,但是这里的log函数并非是log2哦,要记得换底

为了便于理解,我们在后面的代码使用第一种处理方式

给出代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
int n,m,s;
struct edge {
    int next,to;
}mp[2*500000];
int cnt;
int head[500001],fa[500001][20],deep[500001];
int lg[500001];
void add(int x,int y) {
    cnt++;
    mp[cnt].next = head[x];
    mp[cnt].to = y;
    head[x] = cnt;
}
void build(int root,int father) {
    deep[root] = deep[father]+1;
    fa[root][0] = father;
    for(int i=1; i<=lg[deep[root]]+1; i++) {//log+1 +1is necessary!
        fa[root][i] = fa[fa[root][i-1]][i-1]; //important 最重要的一步,相当于ST表中的转移方程
    }
    for(int i=head[root]; ~i; i=mp[i].next) {
        if(mp[i].to!=father) {
            build(mp[i].to,root);    
        }
    }
    return ;
}
int RMQLCA(int x,int y) {
    if(deep[x]<deep[y]) swap(x,y);
    while(deep[x]>deep[y])
        x = fa[x][lg[deep[x]-deep[y]+1]-1];
    if(x==y) return x;
    for(int i=lg[deep[x]]+1-1; i>=0; i--) {//log+1 +1 is necseeary 画图 如果不加,则有的点的祖先会算不到,多加没什么影响,因为仅有f[s][0]=0; 
        if(fa[x][i]!=fa[y][i]) {
            x = fa[x][i];
            y = fa[y][i];    
        }
    }
    return fa[x][0];
} 
int main() {
    int x,y;
    scanf("%d%d%d",&n,&m,&s);
    memset(head,-1,sizeof(head));
    for(int i=1; i<=n-1; i++) {
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    lg[1]=0;
     for(int i = 2; i <= n; ++i)
        lg[i] = lg[i>>1]+1;
//    for(int i = 1; i <= n; ++i)
//        lg[i] = lg[i-1] + (1 << lg[i-1] == i)    
// 也可以直接这样处理为log(x)+1  后面调用就不用+1    
    build(s,0);
    while(m--) {
        scanf("%d%d",&x,&y);
        printf("%d\n",RMQLCA(x,y));
    }
    return 0;
    
}

3、其实求LCA还有一种不常用的方法

Tarjan求LCA

先给出代码:

#include<bits/stdc++.h>
using namespace std;
template<typename Type>inline void read(Type &xx)
{
    Type f=1;char ch;xx=0;
    for(ch=getchar();ch<0||ch>9;ch=getchar())if(ch==-)f=-1;
    for(;ch>=0&&ch<=9;ch=getchar())xx=xx*10+ch-0;
    xx*=f;
}
struct edge
{
    int to,next;
}e[1000001];//边的储存
struct questions
{
    int to,next,same,num;
    bool flag;
    questions(){flag=false;}
}q[1000001];//询问的储存,flag=false表示还没回答,num表示是第几个询问,same储存与这个询问相同的询问序号。
bool b[500001];
int head[500001],que[500001],father[500001];
int n,m,s,nume=0,numq=0,ans[500001];
void add_edge(int x,int y)
{
    e[++nume].to=y;
    e[nume].next=head[x];
    head[x]=nume;
    e[++nume].to=x;
    e[nume].next=head[y];
    head[y]=nume;
}
void add_que(int x,int y,int k)
{
    q[++numq].to=y;
    q[numq].same=numq+1;
    q[numq].next=que[x];
    q[numq].num=k;
    que[x]=numq;
    q[++numq].to=x;//询问要储存到两个点的链表序列里,删的时候也要一起删
    q[numq].same=numq-1;
    q[numq].next=que[y];
    q[numq].num=k;
    que[y]=numq;
}
int find(int x)//并查集
{
    if(father[x]!=x)father[x]=find(father[x]);
    return father[x];
}
void unionn(int x,int y)//并查集
{
    father[find(y)]=find(x);
}
void LCA(int point,int f)//point是当前搜索节点,f是它的父亲
{
    for(int i=head[point];i!=0;i=e[i].next)//遍历与point相连的所有边
        if(e[i].to!=f&&!b[e[i].to])
        {
            LCA(e[i].to,point);
            unionn(point,e[i].to);//合并
            b[e[i].to]=1;
        }
    for(int i=que[point];i!=0;i=q[i].next)//遍历与point相关的询问
        if(!q[i].flag&&b[q[i].to])//如果另一个点遍历过了并且该询问没有回答过
        {
            ans[q[i].num]=find(q[i].to);//记录下答案
            q[i].flag=1;
            q[q[i].same].flag=1;//把两个点上的询问都去掉
        }
}
int main()
{
    read(n);read(m);read(s);
    for(int i=1,x,y;i<=n-1;i++)
    {
        father[i]=i;
        read(x);read(y);
        add_edge(x,y);
    }
    father[n]=n;
    for(int i=1,x,y;i<=m;i++)
    {
        read(x);read(y);
        add_que(x,y,i);
    }
    LCA(s,0);
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
    return 0;
}

具体讲解可看这里

LCA入门

标签:不用   调用   struct   ref   答案   没有   wap   memset   main   

原文地址:https://www.cnblogs.com/delta-cnc/p/12469712.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!