LCA算法——倍增

2021-02-02 20:03:39 阅读：158 来源： 互联网

概况

LCA（Lowest Common Ancestors），即最近公共祖先，是指在有根树中，找出某两个结点u和v最近的公共祖先。

实现过程 预处理：通过dfs遍历，记录每个节点到根节点的距离dist[u]，深度d[u] init()求出树上每个节点u的2^i祖先p[u][i] 求最近公共祖先，根据两个节点的的深度，如不同，向上调整深度大的节点，使得两个节点在同一层上，如果正好是祖先结束，否则，将两个节点同时上移，查询最近公共祖先。代码：

#include<bits/stdc++.h>
#include<queue>
using namespace std;
const int N=5e5+5;
int n,m,s;
int root;
int idx=1;
int ver[2*N],nex[2*N],head[2*N],dis[2*N],f[2*N][25];
queue<int>q;
int t;

void add(int x,int y) {//邻接表建图
    ver[++idx]=y;
    nex[idx]=head[x];
    head[x]=idx;
    return ;
}

void bfs() {//bfs会比dfs更加保险
    q.push(root);
    dis[root]=1;//根节点的深度为1
    while(!q.empty()) {
        int x=q.front();
        q.pop();
        for(int i=head[x]; i; i=nex[i]) {//向下一层搜索
            int y=ver[i];
            if(dis[y]) continue;//防止死循环
            dis[y]=dis[x]+1;//深度更新
            f[y][0]=x;//2^0=1,即为y的父结点
            for(int j=1; j<=t; j++) {
                f[y][j]=f[f[y][j-1]][j-1];//更新2^i个祖先
            }
            q.push(y);
        }
    }
    return ;
}

int lca(int x,int y) {
    if(dis[x]<dis[y]){//严格规范dis[x]>dis[y]
        swap(x,y);    
    } 
    for(int i=t; i>=0; i--) {
        if(dis[f[x][i]]>=dis[y]) x=f[x][i];//使x和y的深度相同
    }
    if(x==y) return x;//已经相同，则直接返回x
    for(int i=t; i>=0; i--) {
        if(f[x][i]!=f[y][i]) {//如果第2^i个祖先不相同，说明还没有到最小公共祖先
            x=f[x][i];
            y=f[y][i];//更新x,y
        }
    }
    return f[x][0];//返回x的父节点——即最小公共祖先
}

int main() {
    cin>>n;
    t=(log(n)/log(2))+1;
    for(int i=1; i<=n; i++) {
        int a,b;
        cin>>a>>b;
        if(b==-1) root=a;
            add(a,b);
            add(b,a);
    }
    bfs();
    cin>>m;
    for(int i=1; i<=m; i++) {
        int x,y;
        cin>>x>>y;
        if(lca(x,y)==x) cout<<1<<endl;
        if(lca(x,y)==y) cout<<2<<endl;
        if(lca(x,y)!=x&&lca(x,y)!=y) cout<<0<<endl;
    }
//    for(int i=1;i<=5;i++)cout<<i<<":"<<dis[i]<<endl;
    return 0;
}

标签：head,idx,int,祖先,算法,LCA,倍增,节点,dis
来源： https://www.cnblogs.com/zhangzhiyuan123/p/14363870.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

LCA算法——倍增

概况