本文最后更新于 191 天前,如有失效请评论区留言。
并查集底层是由森林构成的,森林里的一个树相当于一个集合,并查集可以对集合进行合并(并)和查询(查),“查”的底层是find根节点,目的是区分集合;而“并”就是先找到两个节点的根节点(两个集合),在对根节点进行合并(一般是小树合并到大树),这样就完成了并查集的基本定义。下面是py模板:
# 并查集模板
class UnionFind:
__slots__ = 'fa', 'size'
def __init__(self, n):
self.fa = list(range(n))
self.size = [1] * n
def find(self, x):
if self.fa[x] != x:
self.fa[x] = self.find(self.fa[x])
return self.fa[x]
def merge(self, from_, to):
from_, to = self.find(from_), self.find(to)
if from_ == to:
return False
if self.size[from_] > self.size[to]:
self.fa[to] = from_
self.size[from_] += self.size[to]
else:
self.fa[from_] = to
self.size[to] += self.size[from_]
return True
# 查询并查集联通分块个数
def units_number(self):
return sum(i==x for i, x in enumerate(self.fa))
def is_connected(self, x, y):
return self.find(x) == self.find(y)