【算法笔记】最近公共祖先(LCA)算法求解——倍增算法

news2025/1/10 10:49:45

0. 前言

最近公共祖先简称 LCA(Lowest Common Ancestor)。两个节点的最近公共祖先,就是这两个点的公共祖先里面,离根最远的那个。

这种算法应用很广泛,可以很容易解决树上最短路等问题。

为了方便,我们记某点集 S = { v 1 , v 2 , … , v n } S=\{v_1,v_2,\ldots,v_n\} S={v1,v2,,vn} 的最近公共祖先为 LCA ( v 1 , v 2 , … , v n ) \text{LCA}(v_1,v_2,\ldots,v_n) LCA(v1,v2,,vn) LCA ( S ) \text{LCA}(S) LCA(S)

部分内容参考 OI Wiki,文章中所有算法均使用C++实现。

例题:洛谷 P3379 【模板】最近公共祖先(LCA)

1. 性质

  1. LCA ( { u } ) = u \text{LCA}(\{u\})=u LCA({u})=u
  2. u u u v v v 的祖先,当且仅当 LCA ( u , v ) = u \text{LCA}(u,v)=u LCA(u,v)=u
  3. 如果 u u u 不为 v v v 的祖先并且 v v v 不为 u u u 的祖先,那么 u , v u,v u,v 分别处于 LCA ( u , v ) \text{LCA}(u,v) LCA(u,v) 的两棵不同子树中;
  4. 前序遍历中, LCA ( S ) \text{LCA}(S) LCA(S) 出现在所有 S S S 中元素之前,后序遍历中 LCA ( S ) \text{LCA}(S) LCA(S) 则出现在所有 S S S 中元素之后;
  5. 两点集并的最近公共祖先为两点集分别的最近公共祖先的最近公共祖先,即 LCA ( A ∪ B ) = LCA ( LCA ( A ) , LCA ( B ) ) \text{LCA}(A\cup B)=\text{LCA}(\text{LCA}(A), \text{LCA}(B)) LCA(AB)=LCA(LCA(A),LCA(B))
  6. 两点的最近公共祖先必定处在树上两点间的最短路上;
  7. d ( u , v ) = h ( u ) + h ( v ) − 2 h ( LCA ( u , v ) ) d(u,v)=h(u)+h(v)-2h(\text{LCA}(u,v)) d(u,v)=h(u)+h(v)2h(LCA(u,v)),其中 d d d 是树上两点间的距离, h h h 代表某点到树根的距离。

2. 求解算法

2.0 前置知识1:树的邻接表存储

简单来说,树的邻接表存储就是对于每个结点,存储其能通过一条有向或无向边,直接到达的所有结点。
传统的存储方式是使用链表(或模拟链表),这样实现比较麻烦,也容易写错。
此处为了更好的可读性我们使用STL中的可变长度顺序表vector

#include <vector> // 需要使用STL中的vector
#define maxn 100005 // 最大结点个数

std::vector<int> G[maxn];

此时,若要添加一条无向边 u ↔ v u\leftrightarrow v uv,可使用:

G[u].push_back(v);
G[v].push_back(u);

若要添加 u → v u\to v uv的有向边:

G[u].push_back(v);

遍历 v v v能直接到达的所有结点:

for(int u: G[v])
	cout << u << endl;

2.1 前置知识2:DFS 遍历 & 结点的深度计算

对于两种算法,都需要预处理出每个结点的深度。
一个结点的深度定义为这个结点到树根的距离。

要预处理出所有结点的深度,很简单:
运用树形dp的方法,令 h u h_u hu 表示结点 u u u 的深度,逐层向下推进:

#include <cstdio>
#include <vector>
#define maxn 100005
using namespace std;

vector<int> G[maxn]; // 邻接表存储
int depth[maxn]; // 每个结点的深度

void dfs(int v, int par) // dfs(当前结点,父亲结点)
{
	int d = depth[v] + 1; // 子结点的深度=当前结点的深度+1
	for(int u: G[v])
		if(u != par) // 不加这条判断会无限递归
		{
			depth[u] = d; // dp更新子结点深度
			dfs(u, v); // 往下dfs
		}
}

int main()
{
	// 构建一张图
	// ...
	// 假定图已存入邻接表G:
	int root = 0; // 默认树根为0号结点,根据实际情况设置
	dfs(root, -1); // 对于根结点,父亲结点为-1即为无父亲结点
	return 0;
}

2.2 朴素算法

u , v u,v u,v 表示两个待求 LCA 的结点。需提前预处理出每个结点的父亲(记结点 v v v 的父亲为 f v f_v fv)。

算法步骤:

  1. 使 u , v u,v u,v 的深度相同:可以让深度大的结点往上走,直到与深度小的结点深度相同。
  2. u ≠ v u\ne v u=v时: u ← f u , v ← f v u\gets f_u,v\gets f_v ufu,vfv
  3. 循环直到 u = v u=v u=v,此条件成立后 u u u v v v 的值即为我们要求的 LCA。

时间复杂度分析:

  • 预处理:DFS 遍历整棵树, O ( N ) \mathcal O(N) O(N)
  • 单次查询:最坏 O ( N ) \mathcal O(N) O(N),平均 O ( log ⁡ N ) \mathcal O(\log N) O(logN)(随机树的高为 ⌈ log ⁡ N ⌉ \lceil\log N\rceil logN

参考代码:

#include <cstdio>
#include <vector>
#include <algorithm>
#define maxn 500005
using namespace std;

vector<int> G[maxn];
int depth[maxn], par[maxn];

void dfs(int v)
{
	int d = depth[v] + 1;
	for(int u: G[v])
		if(u != par[v])
		{
			par[u] = v, depth[u] = d;
			dfs(u);
		}
}

int lca(int u, int v)
{
	if(depth[u] < depth[v])
		swap(u, v);
	while(depth[u] > depth[v])
		u = par[u];
	while(u != v)
		u = par[u], v = par[v];
	return u;
}

int main()
{
	int n, q, root;
	scanf("%d%d%d", &n, &q, &root);
	for(int i=1; i<n; i++)
	{
		int u, v;
		scanf("%d%d", &u, &v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	par[root] = -1, depth[root] = 0;
	dfs(root);
	while(q--)
	{
		int u, v;
		scanf("%d%d", &u, &v);
		printf("%d\n", lca(u, v));
	}
	return 0;
}

可以发现,程序在最后四个测试点上TLE了:

TLE

这是因为,这四个点是专门针对朴素算法设计的(正好是一个 Subtask),使算法的时间复杂度达到了最坏情况 O ( N Q ) \mathcal O(NQ) O(NQ),而 N , Q ≤ 5 × 1 0 5 N,Q\le 5\times 10^5 N,Q5×105,所以无法通过测试点。当然,朴素算法在随机树上回答 Q Q Q 次询问的时间复杂度还是 O ( N + Q log ⁡ N ) \mathcal O(N+Q\log N) O(N+QlogN),被极端数据卡掉也没办法

2.3 倍增

倍增算法是朴素算法的改进算法,也是最经典的 LCA 求法。

预处理:

  • fa x , i \text{fa}_{x,i} fax,i 表示点 x x x 的第 2 i 2^i 2i 个祖先。
  • dfs 预处理深度信息时,也可以预处理出 fa x , i \text{fa}_{x,i} fax,i
    • 首先考虑 i i i的范围: 2 i ≤ d x 2^i\le d_x 2idx(前面说的, d x d_x dx 表示结点 x x x 的深度),所以有 0 ≤ i ≤ ⌊ log ⁡ 2 d x ⌋ 0\le i\le \lfloor\log_2 d_x\rfloor 0ilog2dx
    • 对于 i = 0 i=0 i=0 2 i = 2 0 = 1 2^i=2^0=1 2i=20=1,所以直接令 fa x , 0 = ( x 的父亲 ) \text{fa}_{x,0}=(x\text{的父亲}) fax,0=(x的父亲) 即可。
    • 对于 1 ≤ i ≤ ⌊ log ⁡ 2 d x ⌋ 1\le i\le \lfloor\log_2 d_x\rfloor 1ilog2dx x x x 的第 2 i 2^i 2i 个祖先可看作 x x x 的第 2 i − 1 2^{i-1} 2i1 个祖先的第 2 i − 1 2^{i-1} 2i1 个祖先( 2 i − 1 + 2 i − 1 = 2 i 2^{i-1}+2^{i-1}=2^i 2i1+2i1=2i),即:
      fa x , i = fa fa x , i − 1 , i − 1 \text{fa}_{x,i}=\text{fa}_{\text{fa}_{x,i-1},i-1} fax,i=fafax,i1,i1

求解步骤:

  1. 使 u , v u,v u,v 的深度相同:计算出 u , v u,v u,v 两点的深度之差,设其为 y y y。通过将 y y y 进行二进制拆分,我们将 y y y 次游标跳转优化为「 y y y 的二进制表示所含 1 的个数」次游标跳转(详见代码)。
  2. 特判:如果此时 u = v u=v u=v,直接返回 u u u v v v 作为 LCA 结果。
  3. 同时上移 u u u v v v:从 i = ⌊ log ⁡ 2 d u ⌋ i=\lfloor\log_2 d_u\rfloor i=log2du 开始循环尝试,一直尝试到 0 0 0(包括 0 0 0),如果 fa u , i ≠ fa v , i \text{fa}_{u,i}\not=\text{fa}_{v,i} fau,i=fav,i,则 u ← fa u , i , v ← fa v , i u\gets\text{fa}_{u,i},v\gets\text{fa}_{v,i} ufau,i,vfav,i,那么最后的 LCA 为 fa u , 0 \text{fa}_{u,0} fau,0

时间复杂度分析:

  • 预处理: O ( N ) \mathcal O(N) O(N) DFS ×   O ( log ⁡ N ) \times~\mathcal O(\log N) × O(logN) 预处理 = O ( N log ⁡ N ) =\mathcal O(N\log N) =O(NlogN)
  • 单次查询:平均 O ( log ⁡ N ) O(\log N) O(logN),最坏 O ( log ⁡ N ) O(\log N) O(logN)
  • 预处理 + Q Q Q 次查询: O ( N + Q log ⁡ N ) \mathcal O(N+Q\log N) O(N+QlogN)

另外倍增算法可以通过交换 fa 数组的两维使较小维放在前面。这样可以减少 cache miss 次数,提高程序效率。

参考代码:

#include <cstdio>
#include <vector>
#include <cmath>
#define maxn 500005
using namespace std;

vector<int> G[maxn];
int fa[maxn][19]; // 2^19=524288
int depth[maxn];

void dfs(int v, int par)
{
	fa[v][0] = par;
	int d = depth[v] + 1;
	for(int i=1; (1<<i)<d; i++)
		fa[v][i] = fa[fa[v][i - 1]][i - 1];
	for(int u: G[v])
		if(u != par)
			depth[u] = d, dfs(u, v);
}

inline int lca(int u, int v)
{
	if(depth[u] < depth[v])
		u ^= v ^= u ^= v;
	int m = depth[u] - depth[v];
	for(int i=0; m; i++, m>>=1)
		if(m & 1)
			u = fa[u][i];
	if(u == v) return u; // 这句不能丢
	for(int i=log2(depth[u]); i>=0; i--)
		if(fa[u][i] != fa[v][i])
			u = fa[u][i], v = fa[v][i];
	return fa[u][0];
}

int main()
{
	int n, q, root;
	scanf("%d%d%d", &n, &q, &root);
	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);
	}
	depth[--root] = 0;
	dfs(root, -1);
	while(q--)
	{
		int u, v;
		scanf("%d%d", &u, &v);
		printf("%d\n", lca(--u, --v) + 1);
	}
	return 0;
}

AC

3. 习题

  • 题目链接:洛谷 P8805 [蓝桥杯 2022 国 B] 机房
  • 题解:https://best-blogs.blog.luogu.org/solution-p8805

4. 总结

本文详细讲解了 LCA 问题以及求解 LCA 的两种算法。对比如下:

算法预处理时间复杂度单次查询时间复杂度1空间复杂度能否通过例题2
朴素算法 O ( N ) \mathcal O(N) O(N) O ( N ) \mathcal O(N) O(N) O ( N ) \mathcal O(N) O(N)
倍增算法 O ( N log ⁡ N ) \mathcal O(N\log N) O(NlogN) O ( log ⁡ N ) \mathcal O(\log N) O(logN) O ( N log ⁡ N ) \mathcal O(N\log N) O(NlogN)✔️

创作不易,希望大家能给个三连,感谢支持!


  1. 此时间复杂度按照最坏情况计算。 ↩︎

  2. 例题:洛谷 P3379 【模板】最近公共祖先(LCA) ↩︎

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/146460.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

再获认可!知道创宇入选《2022中国各地区科创之星势力图3.0版》

近日&#xff0c;由数据智能产业创新服务领域的垂直、专业深度媒体数据猿推出的《2022中国各地区科创之星势力图3.0版》正式公布。知道创宇凭借网络安全领域过硬的技术实力与创新能力入选“科创之星势力图”。二度认可知道创宇再获科创之星优秀企业本次数据猿公布的《2022中国各…

第十九章AJAX学习

文章目录什么是AJAX服务器端渲染Ajax渲染&#xff08;局部更新&#xff09;前后端分离同步与异步同步异步前端提交到数据到后端的方式JSON什么是JSONJSON数据格式JSON对象JSON格式表示简单数组对象数组使用二维数组保存对于客户端对于服务器端引入GSON的jar包使用例子Axios简介…

MySQL数据库:库和表的基本操作

SQL注意事项 1.每条sql语句都要以英文分号结尾。 2.sql语句不区分大小写。 3.定义库、表&#xff0c;或是表中的字段&#xff0c;名称不要使用关键字。 一、数据库的操作 1.显示所有的数据库 show databases; 2.创建数据库 方式一&#xff1a; create database 数据库名称…

vue+elementui导入Excel文件(基于vue-element-admin中的uploadExcel组件), 然后 go-zero进行逻辑处理功能

#前端代码 html代码 <el-button v-waves class"filter-item" type"primary" icon"add" click"downLoadExlce">模板下载</el-button> <el-upload class"upload" action :multiple"false" :show-…

2.两数相加(链表)

题目 给你两个 非空 的链表&#xff0c;表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的&#xff0c;并且每个节点只能存储 一位 数字。 请你将两个数相加&#xff0c;并以相同形式返回一个表示和的链表。 你可以假设除了数字 0 之外&#xff0c;这两个数都不会…

【Linux】动静态库的制作和使用

目录 一、可执行文件的形成 二、静态库 2.1 制作静态库 2.2 使用静态库 三、动态库 3.1 动态库的制作 3.2 动态库链接 2.3 动态库的加载 一、可执行文件的形成 在C语言中&#xff0c;函数库文件分为两种类型&#xff1a; 静态库&#xff08;库程序是直接注入目标程序…

想要做成一件事,就得有方法,怎么做到呢

想要完成一件事&#xff0c;没有详细的计划和方法是做不好的。如果没有计划&#xff0c;没有方案&#xff0c;没有策略&#xff0c;没有规划&#xff0c;没有方法&#xff0c;那就是一只无头苍蝇&#xff0c;到处乱撞&#xff0c;不能达到什么目的。想做成一件事&#xff0c;就…

后端人眼中的Vue(二)

三、Vue基本语法 3.1、展示数据 3.1.1、v-text 3.1.1.1、简介 ​ 和插值表达式&#xff08;{{}}&#xff09;一样&#xff0c;用于获取vue实例中data属性声明的数据。用法是在哪个标签上获取就直接在哪个标签上定义v-text或者是v-html。相当于javascript中的innerText。直接…

笔记:Android 系统架构

Android系统架构 1.应用层&#xff08;System Apps&#xff09; 系统内置应用程序和非系统应用程序&#xff0c;源码目录中的 packages 目录对应系统应用层 2.应用架构层&#xff08;Framework&#xff09; 为开发人员提供应用程序开发所需要的API&#xff0c;向下与c/c程序…

J-Tech Talk|跨模态视频检索进阶,一起探索CLIP模型的新天地!

J-Tech Talk由 Jina AI 社区为大家带来的技术分享围绕 Python 的相关话题工程师们将深入细节地讲解具体的问题分享 Jina AI 在开发过程中所积累的经验CLIP 模型在零样本图像分类、跨模态检索中效果拔群&#xff0c;它的出现同时推动了 NLP 和 CV 的发展&#xff0c;为解决许多实…

【自学C++】C++转义字符

C转义字符 C转义字符教程 在编程中有一些字符是打不出来的&#xff0c;比如换行&#xff0c;提示音&#xff0c;下一个制表位等等&#xff0c;于是程序语言的开发者就设计了转义序列&#xff08;escape sequence&#xff09;这种东西用来表达这些字符。 所谓转义&#xff0c…

jvm学习的核心(二)---运行时数据区概述

文章目录1.方法区&#xff08; Method Area&#xff09;2.堆&#xff08;Heap&#xff09;3.本地方法栈&#xff08;Native Method Stacks&#xff09;4.程序计数器&#xff08;pc Register&#xff09;5.虚拟机栈&#xff08;Virtual Machine Stacks&#xff09;运行时数据区 …

IDEA启动项目通过https进行访问,nginx配置https访问

一、IDEA启动项目通过https进行访问 1、获取证书 证书申请网络有很多方法&#xff0c;这里腾讯云的nginx版证书举列 2、证书转换 springboot是识别p12证书的&#xff0c;所有这里需要吧pem证书进行转换。转换工具OpenSSL-Win64&#xff08;自行百度下载&#xff09;下载完…

nodejs操作MySQL数据库

1、操作数据库的步骤 安装操作 MySQL 数据库的第三方模块&#xff08;mysql&#xff09;通过 mysql 模块连接到 MySQL 数据库通过 mysql 模块执行 SQL 语句2、安装与配置 mysql 模块 2.1、安装 mysql 模块 mysql 模块是托管于 npm 上的第三方模块。它提供了在 Node.js 项目中…

【Centos】服务管理、解/压缩、磁盘、进程管理相关命令

文章目录一、服务管理1 service2 chkconfig设置后台服务器的自启配置3 systemctl设置后台服务器自启配置防火墙关闭4 开关机重启5 搜索查找类6 locate快速定位文件路径二、压缩、解压1 gzip2 zip3 tar三、磁盘查看和分区类1 du2 df3 lsblk4 mount5 fdisk四、进程管理类1 iotop2…

Vue基础入门小demo——图片切换(初阶)

文章目录 &#x1f4cb;前言 &#x1f3af;demo介绍 &#x1f3af;完整代码 &#x1f3af;最终效果 &#x1f3af;案例解析 &#x1f4cb;前言 图片切换是一个很经典的Vue入门学习案例&#xff0c;在你学习完一些基本的v-指令后&#xff0c;你可以尝试去写一个简单的demo去…

吴恩达【神经网络和深度学习】Week3——浅层神经网络

文章目录Shallow Neural Network1、Neural Networks Overview2、Neural Network Representation3、Computing a Neural Networks Output4、Vectorizing Across Multiple Examples5、Explanation for Vectorized Implementation6、Activation Functions7、Why do you need Non-L…

年底固定资产盘点的正确招数

企业年终盘点的必要性 年终盘点的最终目的是摸清资产现状&#xff0c;掌握资产情况&#xff0c;为日常管理和来年采购预算做有力数据支撑&#xff0c;指导我们接下来应该怎么去做&#xff0c;对任何一家企业的管理都有着重要意义。 随着企业规模的扩大&#xff0c;员工和企业设…

SCI论文解读复现【NO.4】FINet:基于合成雾和改进YOLOv5的绝缘子数据集和检测基准(代码已复现)

此前出了目标检测算法改进专栏&#xff0c;但是对于应用于什么场景&#xff0c;需要什么改进方法对应与自己的应用场景有效果&#xff0c;并且多少改进点能发什么水平的文章&#xff0c;为解决大家的困惑&#xff0c;此系列文章旨在给大家解读发表高水平学术期刊中的SCI论文&am…

Redisson获取/释放分布式锁流程中使用的方法以及watchDog机制相关源码分析

Redisson获取/释放分布式锁原理以及watchDog机制相关源码分析使用到的重点类继承结构RedissonLockExpirationEntry获取锁的代码逻辑tryLock()tryLock(long waitTime, long leaseTime, TimeUnit unit)tryAcquire(long waitTime, long leaseTime, TimeUnit unit, long threadId)t…