当前位置: 首页 > news >正文

D006 【模板】并查集

并查集是非常灵活和高效的数据结构,常见应用是维护无向图的连通分量个数、大小,最小生成树的 Kruskal 算法和最近公共祖先等。

并查集维护了若干个不相交的集合,每个集合通过一棵树来组织,根节点为该集合的代表。

三个基本操作:

  1. init(n) :初始化含有 \(n\) 个集合的并查集,每个集合的代表为自身。
  2. find(x) :寻找元素 \(x\) 所在集合的代表。
  3. union(x, y) :将元素 \(x,y\) 所在的集合合并,如果已经是处于同一集合的话就不合并。

一个简单的并查集示例,可以先看图理解,再理解代码结构。

并查集初始化、合并即查找图例

图示的一个关键理解是:当 fa[x] == x 时,说明 x 是根节点。

def __init__(self, n):self.fa = list(range(n + 1))

find 函数查找到元素所在集合的根,并且在查找过程中进行路径压缩(Path Compression)。

路径压缩目的是把树的高度压低,使得长链可以变成扁平的放射状,从而大大降低时间复杂度。

递归写法(直观、适合带权并查集写法)

def find(self, x):if self.fa[x] == x:return xroot = self.find(self.fa[x]) self.fa[x] = root return root

非递归写法(Python 推荐,避免栈溢出)

这里采用路径减半策略,效率高且代码简洁。

def find(self, x):while self.fa[x] != x:self.fa[x] = self.fa[self.fa[x]]x = self.fa[x]return x

在查找时当自身不是根时往上跳,跳到根节点,然后再返回的过程中改变途径节点的父节点,统一改成根。

合并的话先使用 find 函数找到对应的根,再链接即可。

def union(self, x, y):rx = self.find(x)ry = self.find(y)if rx != ry:self.fa[rx] = ry  # 将 rx -> ry

这里也可以使用一个简单的启发式策略——按秩合并。维护一个 size 数组把小集合挂到大集合里去。

def union(self, x, y):rx = self.find(x)ry = self.find(y)if rx != ry:self.fa[rx] = ry  # 将 rx -> ryif size[rx] > size[ry]:rx, ry = ry, rxself.fa[rx] = rysize[ry] += size[rx]

具体模板为

class DSU:def __init__(self, n):self.fa = list(range(n + 1))def find(self, x):while self.fa[x] != x:self.fa[x] = self.fa[self.fa[x]]x = self.fa[x]return xdef union(self, x, y):rx = self.find(x)ry = self.find(y)if rx != ry:self.fa[rx] = rydef is_same(self, x, y):rx = self.find(x)ry = self.find(y)return rx == ry
http://www.jsqmd.com/news/436755/

相关文章:

  • 别错过!AI应用架构师阐述AI驱动虚拟世界构建新策略
  • 2026成人教育本科推荐:上班族学历含金量与毕业通过率十家机构深度评测 - 速递信息
  • 2026年3月片材机组厂家推荐榜:甄选企业实测解析 - 品牌鉴赏师
  • 2026年3月smc片材厂家推荐,行业权威盘点与品质红榜发布 - 品牌鉴赏师
  • 零基础必备!TOP5手机公众号排版工具推荐 微信图文编辑选择指南 - 速递信息
  • Flink如何提升大数据领域的数据处理效率
  • 中望3D2026曲线合并(连接)操作指南
  • 残差突破的机缘巧合(五,cudnn残差类层改正)
  • 【2026最新】Balabolka下载汉化版:最强文本转语音工具(附安装包+图文安装步骤) - xiema
  • 2026年3月C型斗式提升机厂家最新推荐,大流量平稳输送实力厂家 - 品牌鉴赏师
  • Ubuntu 22.04 安装与更新 OpenSpec 教程(含 nvm / Node.js)
  • 想考成人大专不知怎么选?2026十家高通过率机构学费与学制对比 - 速递信息
  • 前端接私活必看:XinServer 提速到底有多夸张?
  • goGorm不更新0值?
  • C++游戏开发之旅 23
  • gorm save 修改时非空字段不保存!
  • P12742 [POI 2016 R3] 信使 Messenger
  • 从0到1吃透Agent、MCP、Skills的关系!
  • 京东e卡回收新思路,解锁变现新姿势 - 京顺回收
  • ComPDF的产品升级:从工具包到PDF服务 - 实践
  • 2026年3月连斗式提升机厂家最新推荐,连续上料效率更高 - 品牌鉴赏师
  • 2026年3月仿大理石板材设备厂家推荐,行业权威盘点与品质红榜发布 - 品牌鉴赏师
  • 第一类斯特林数列
  • 2026年工程铺路钢板出租,优质厂家助力项目,工地施工钢板出租/临时道路钢板出租,工程铺路钢板出租厂家哪个好 - 品牌推荐师
  • 2026年中国露点仪市场白皮书:知名厂家推荐与高精度监测技术深度对标
  • 2026年3月管道涂塑设备厂家推荐,行业测评与采购选择指南 - 品牌鉴赏师
  • 2026年3月钢管粉末喷涂设备厂家最新推荐,粉末涂装技术实力优选 - 品牌鉴赏师
  • 2025年12月GESP真题及题解(C++七级): 选择题和判断题(题解)
  • 站内Geo优化SOP:专家于磊“两大核心+四轮驱动”实战指南
  • 大模型(LLMs)从入门到精通:涵盖基础、进阶、微调、LangChain及参数高效微调全解析!