生成树的概念就是满足有n个顶点的生成树,有且仅有n-1条边,但是有n-1条边的图不一定都是生成树。准则:
1、必须只使用该图的边来构造最小生成树
2、必须使用且仅使用n-1条边来连接图中的n个顶点
3、不能使用产生回路的边。
保证任意两个节点都可以互相链接就可以。
很简单。直接用并查集完事。先排序,然后并查集查询是否是同个根节点,如果是,那么肯定不可以连接,否则回路生成。
所以 并查集一个关键作用就是可以查询是否存在联系。
class Solution:
def miniSpanningTree(self , n , m , cost ):
# write code here
if not cost:
return 0
record = {}
cost = sorted(cost, key = lambda x:x[2])
result = 0
def find(x):
record.setdefault(x, x)
while record[x] != x:
x = record[x]
return x
def union(tmp1, tmp2):
record[tmp1] = tmp2
for i in range(len(cost)):
res1, res2, weight = cost[i]
tmp1, tmp2 = find(res1), find(res2)
if tmp1 != tmp2:
result += weight
union(tmp1, tmp2)
return result