Python高级数据结构——并查集（Disjoint Set）

Python中的并查集（Disjoint Set）：高级数据结构解析

并查集是一种用于处理集合的数据结构，它主要支持两种操作：合并两个集合和查找一个元素所属的集合。在本文中，我们将深入讲解Python中的并查集，包括并查集的基本概念、实现方式、路径压缩和应用场景，并使用代码示例演示并查集的操作。

基本概念

1. 并查集的表示

并查集通常使用树来表示集合，其中每个节点表示一个元素，树的根节点表示集合的代表元素。

class DisjointSet:def __init__(self, size):self.parent = [i for i in range(size)]self.rank = [0] * sizedef find(self, x):if self.parent[x] != x:self.parent[x] = self.find(self.parent[x])  # 路径压缩return self.parent[x]def union(self, x, y):root_x = self.find(x)root_y = self.find(y)if root_x != root_y:if self.rank[root_x] < self.rank[root_y]:self.parent[root_x] = root_yelif self.rank[root_x] > self.rank[root_y]:self.parent[root_y] = root_xelse:self.parent[root_x] = root_yself.rank[root_y] += 1# 示例
disjoint_set = DisjointSet(5)
disjoint_set.union(0, 1)
disjoint_set.union(1, 2)
disjoint_set.union(3, 4)

2. 路径压缩

路径压缩是通过在 find 操作中将节点直接连接到根节点来优化并查集的性能。它减小了树的高度，使得后续的 find 操作更快。

def find(self, x):if self.parent[x] != x:self.parent[x] = self.find(self.parent[x])  # 路径压缩return self.parent[x]

应用场景

并查集常用于解决集合的合并和查找问题，例如：

网络连接问题：判断网络中的节点是否连通。
社交网络中的关系：判断两个人是否属于同一个社交圈。
图的连通性问题：判断图中的节点是否在同一个连通分量中。

代码示例：解决网络连接问题

def are_nodes_connected(disjoint_set, node1, node2):return disjoint_set.find(node1) == disjoint_set.find(node2)# 示例
disjoint_set_network = DisjointSet(10)
disjoint_set_network.union(0, 1)
disjoint_set_network.union(1, 2)
disjoint_set_network.union(3, 4)print(are_nodes_connected(disjoint_set_network, 0, 2))  # 输出: True
print(are_nodes_connected(disjoint_set_network, 0, 3))  # 输出: False