倍增算法 最近公共祖先(LCA)C++

倍增算法(Binary Lifting)是一种用于解决最近公共祖先(LCA)问题的算法。

最近公共祖先问题是指在一棵有根树中,给定两个节点,找到它们的最近公共祖先节点。最近公共祖先节点是指在树中同时是这两个节点的祖先节点,并且离这两个节点的距离最近。

倍增算法的核心思想是在树的深度方向上使用二进制的方式进行跳跃。具体步骤如下:

  1. 预处理:对于每个节点,计算它的2^i级祖先节点,其中i从0开始递增。这可以通过自底向上的动态规划来实现。第一级祖先节点是每个节点的父节点,第二级祖先节点是每个节点的父节点的父节点,以此类推。

  2. 查询:给定两个节点x和y,首先判断它们的深度d,将深度较大的节点向上移动到与另一个节点处于同一深度。然后,从最大的2^i开始,每次向上跳跃2^i个级别,直到两个节点达到同一级别。在跳跃过程中,每次都检查是否达到了公共祖先节点。每个节点跳跃的级别数取决于两个节点的深度差。

倍增算法的时间复杂度为O(log n),其中n为树的节点数。预处理需要O(n log n)的时间来计算每个节点的祖先节点,查询阶段需要O(log n)的时间来计算最近公共祖先节点。空间复杂度为O(n log n),用于存储每个节点的祖先节点。

复制代码
#include <bits/stdc++.h>
#define ll long long
#define endl "\n"
#define KUI ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
using namespace std;
const int con = 5e5 + 5;
const int mod = 998244353;
int n, m, s, k, dep[con], fa[con][20];
vector<int> v[con];
void dfs(int u, int f)
{
    dep[u] = dep[f] + 1;
    fa[u][0] = f;
    for (int i = 1; i <= 19; i++)
    {
        fa[u][i] = fa[fa[u][i - 1]][i - 1];
    }
    for (int v : v[u])
    {
        if (v != f)
        {
            dfs(v, u);
        }
    }
}
int lca(int u, int v)
{
    if (dep[u] < dep[v])
    {
        swap(u, v);
    }
    for (int i = 19; i >= 0; i--)
    {
        if (dep[fa[u][i]] >= dep[v])
        {
            u = fa[u][i];
        }
    }
    if (u == v)
    {
        return v;
    }
    for (int i = 19; i >= 0; i--)
    {
        if (fa[u][i] != fa[v][i])
        {
            u = fa[u][i], v = fa[v][i];
        }
    }
    return fa[u][0];
}
int main()
{
    KUI;
    cin >> n >> m >> s;
    for (int i = 1; i <= n - 1; i++)
    {
        int a1, a2;
        cin >> a1 >> a2;
        v[a1].push_back(a2);
        v[a2].push_back(a1);
    }
    dfs(s, 0);
    for (int i = 1; i <= m; i++)
    {
        int a1, a2;
        cin >> a1 >> a2;
        cout << lca(a1, a2) << endl;
    }
    return 0;
}
相关推荐
凌肖战19 分钟前
力扣网编程55题:跳跃游戏之逆向思维
算法·leetcode
念九_ysl33 分钟前
Java 使用 OpenHTMLToPDF + Batik 将含 SVG 遮罩的 HTML 转为 PDF 的完整实践
java·开发语言·pdf
yaoxin52112343 分钟前
124. Java 泛型 - 有界类型参数
java·开发语言
liulilittle1 小时前
深度剖析:OPENPPP2 libtcpip 实现原理与架构设计
开发语言·网络·c++·tcp/ip·智能路由器·tcp·通信
88号技师1 小时前
2025年6月一区-田忌赛马优化算法Tianji’s horse racing optimization-附Matlab免费代码
开发语言·算法·matlab·优化算法
勤奋的知更鸟1 小时前
Java 编程之模板方法模式
java·开发语言·模板方法模式
ゞ 正在缓冲99%…2 小时前
leetcode918.环形子数组的最大和
数据结构·算法·leetcode·动态规划
十年编程老舅2 小时前
跨越十年的C++演进:C++20新特性全解析
c++·c++11·c++20·c++14·c++23·c++17·c++新特性
碎叶城李白2 小时前
若依学习笔记1-validated
java·笔记·学习·validated
上单带刀不带妹2 小时前
手写 Vue 中虚拟 DOM 到真实 DOM 的完整过程
开发语言·前端·javascript·vue.js·前端框架