并查集的思想就是“认大哥”
一开始默认的大哥都是自己,后来如果遇到可以合并的条件,则将其中一个的大哥改成另个一个大哥,即其中一个区域的人全部转投另一个区域的大哥,最后的集合数,即自己是自己的大哥的个数(自己跟别人混,则证明自己处在一个连通域中,不用计数)
parent = list(range(n)) #n为长度
def find(index):
if index != parent[index]:
parent[index] = find(parent[index])
return parent[index]
def union(index1,index2):
parent[find(index1)] = find(index2)
for i in range(n):
for j in range(i):
if (): #加入并查集条件
union(i,j)
print(sum([i==parent[i] for i in range(n)])) #连通域数
由于在更新时的条件问题,例如:连接[0,3],[1,2], [0,2]
按照道理来说这四个点都应同属一个区域,但是在parent标记时会出现[0,0,1,0]的情况,这个parent所代表的的意思就是同一区域,但是使用的时候会带来不便,因此可以进行统一化操作
dic = {index:value for index,value in enumerate(parent)}
for i in range(n):
while(parent[i] != dic[parent[i]]):
parent[i] = dic[parent[i]]
or
for i in range(n):
parent[i] = find(parent[i])
统一成[0,0,0,0]