最近公共祖先(LCA)//模板

题目描述
如题,给定一棵有根多叉树,请求出指定两个点直接最近的公共祖先。
输入格式
第一行包含三个正整数N、M、S,分别表示树的结点个数、询问的个数和树根结点的序号。
接下来N-1行每行包含两个正整数x、y,表示x结点和y结点之间有一条直接连接的边(数据保证可以构成树)。
接下来M行每行包含两个正整数a、b,表示询问a结点和b结点的最近公共祖先。
输出格式
输出包含M行,每行包含一个正整数,依次为每一个询问的结果。

基于倍增思想求LCA步骤:

1.先预处理出各节点的深度和他们2^j的级的祖先
2.预处理出log2(i)+1的值
3.计算lca

#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
typedef long long ll;
const int maxn=5e5+7;
const int inf=1e7+7;
int n,m,s;
vector<int> g[maxn];
int dep[maxn],f[maxn][22],lg[maxn];
//f[i][j]表示i往上走2^j步到达的点
void dfs(int u,int fa){
    dep[u]=dep[fa]+1;
    f[u][0]=fa;
    for(int j=1;(1<<j)<=dep[u];j++)//u的2^(j-1)祖先的2^(j-1)祖先
        f[u][j]=f[f[u][j-1]][j-1];
    for(int i=0;i<(int)g[u].size();i++){
        int v=g[u][i];
        if(v!=fa)dfs(v,u);
    }
}
void init(){
    for(int i=1;i<=n;i++)//预处理出log2(i)+1的值,便于直接调用
        lg[i]=lg[i-1]+(1<<lg[i-1]==i);
}
int lca(int u,int v){
    if(dep[u]<dep[v])swap(u,v);//不妨设u的深度 >= v的深度,便于下面统一处理
    while(dep[u]>dep[v])u=f[u][lg[dep[u]-dep[v]]-1];//跳到统一深度
    if(u==v)return u;
    for(int j=lg[dep[u]]-1;j>=0;j--)//从步数大往小的跳
        if(f[u][j]!=f[v][j])//因为我们要跳到它们LCA的下面一层,所以它们肯定不相等,如果不相等就跳过去
            u=f[u][j],v=f[v][j];
    return f[u][0];//返回父节点
}
int main(){
    scanf("%d%d%d",&n,&m,&s);
    for(int i=0;i<n-1;i++){
        int u,v;scanf("%d%d",&u,&v);
        g[u].push_back(v);
        g[v].push_back(u);
    }
    dfs(s,0);
    init();
    for(int i=0;i<m;i++){
        int u,v;scanf("%d%d",&u,&v);
        printf("%d\n",lca(u,v));
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值