集合使用双亲表示法,给一个结点,找到它的根节点,即它所在的集合
集合常用方法:并、查找
路径压缩:一次查找就把处在一链上的结点的父结点,直接指向集合的根节点
按秩归并:比较两个集合的规模,或者树的高度,
永远让小树并到大树上,维持树的高度
class SetADT:
def __init__(self):
self.set = list() # 默认集合初始元素为-1
def find(self, target):
if self.set[target] < 0: # 找到集合的根
return target
else:
# 先找到根,把根变成 x 的父结点,再返回根
self.set[target] = self.find(self.set[target]) # 路径压缩
return self.set[target]
def union(self, par1=None, par2=None):
if self.set[par1] < self.set[par2]: # 按秩归并
self.set[par1] += self.set[par2]
self.set[par2] = par1
else:
self.set[par2] += self.set[par1]
self.set[par1] = par2
def test_set():
set_1 = SetADT()
set_1.set = [6, 6, -4, 4, 2, 2, -3] # 根来表示集合的规模
print(set_1.find(3))
set_1.union(2,6)
print(set_1.set[6])