最小生成数

生成树的概念就是满足有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
    
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值