最近公共祖先(lca)倍增算法【模板】

P3379 【模板】最近公共祖先(LCA) - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

复制代码
#include<bits/stdc++.h>
#include<cstdio>
using namespace std;
const int N=5e5+100;
const int inf=0x3f3f3f;
int n,m,s;
vector<int>g[N];
int dep[N];//存u点的深度
int fa[N][20];//存从u点向上跳2^i层的祖先节点
void dfs(int u,int father)
{
    dep[u]=dep[father]+1;//该点深度为他的父节点深度加1
    fa[u][0]=father;//当前节点向上跳一步就是他的父节点
    for(int i=1;i<=19;i++)
        fa[u][i]=fa[fa[u][i-1]][i-1];//递推点的所有能跳到的祖先节点,越界为0
    for(auto i:g[u])
        if(i!=father)
            dfs(i,u);
}
int lca(int u,int v)
{
    if(dep[u]<dep[v])//让u点跳
        swap(u,v);
    /*for(int i=19;i>=0;i--)//先让两点到达同一层
    {
        if(dep[fa[u][i]]>=dep[v])//u节点跳到祖先节点后的深度大于v点的深度
            u=fa[u][i];//u点变到其祖先节点处
    }*/
    int dx=dep[u]-dep[v];
    for(int i=0;dx>0;i++,dx=dx/2)
    {
        if(dx&1)
        {
            u=fa[u][i];
        }
    }
    //两种写法,让两点到达同一层
    
    if(u==v)//特判,v点为u的祖先节点
        return v;
    for(int i=19;i>=0;i--)
    {
        if(fa[u][i]!=fa[v][i])//两点跳后的祖先节点的父节点是否相等,不相等i-1
            u=fa[u][i],v=fa[v][i];
        //停在最近公共祖先的下层的儿子节点(此时两点的父节点就是公共最近祖先)
    }
    return fa[u][0];//再跳最后一步到达公共祖先位置
}
int main()
{
    scanf("%d%d%d",&n,&m,&s);
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs(s,0);//从根节点遍历树
    while(m--)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        printf("%d\n",lca(a,b));
    }
    return 0;
}
相关推荐
先做个垃圾出来………26 分钟前
差分数组(Difference Array)
java·数据结构·算法
hansang_IR1 小时前
【题解】洛谷 P4286 [SHOI2008] 安全的航线 [递归分治]
c++·数学·算法·dfs·题解·向量·点积
乐迪信息1 小时前
乐迪信息:AI摄像机在智慧煤矿人员安全与行为识别中的技术应用
大数据·人工智能·算法·安全·视觉检测
多恩Stone1 小时前
【3DV 进阶-2】Hunyuan3D2.1 训练代码详细理解下-数据读取流程
人工智能·python·算法·3d·aigc
惯导马工2 小时前
【论文导读】IDOL: Inertial Deep Orientation-Estimation and Localization
深度学习·算法
老姜洛克2 小时前
自然语言处理(NLP)之n-gram从原理到实战
算法·nlp
CoovallyAIHub3 小时前
基于YOLO集成模型的无人机多光谱风电部件缺陷检测
深度学习·算法·计算机视觉
CoovallyAIHub3 小时前
几十个像素的小目标,为何难倒无人机?LCW-YOLO让无人机小目标检测不再卡顿
深度学习·算法·计算机视觉
怀旧,3 小时前
【C++】19. 封装红⿊树实现set和map
linux·c++·算法
往事随风去3 小时前
Redis的内存淘汰策略(Eviction Policies)有哪些?
redis·后端·算法