题解 | #最小生成树#

最小生成树

https://www.nowcoder.com/practice/735a34ff4672498b95660f43b7fcd628

# 返回最小的花费代价使得这n户人家连接起来
# @param n int整型 n户人家的村庄
# @param m int整型 m条路
# @param cost int整型二维数组 一维3个参数,表示连接1个村庄到另外1个村庄的花费的代价
# @return int整型

# 最小生成树: 最小生成树包含原图的所有节点而只用最少的边和最小的权值距离
# 视频学习:https://www.bilibili.com/video/BV1d34y1R7xA/?spm_id_from=333.337.search-card.all.click&vd_source=e0d139ef9278fd07bd20752f3c433bb0
# 算法思想 :将所有边的权值按照从小到大做升序排序,从最小的开始选择,新加入的只要不与原来的边构成环路,就可加入到最小生成树的
    # • 树中没有环
    # • 所有的顶点都要在树中
    # • 对于N条顶点,有N-1条边

class Solution:
    def miniSpanningTree(self , n , m, cost):
        'n个顶点, m条边'
        '首先对图中的边按照边权从小到大排序'
        cost=sorted(cost, key=lambda x: x[2])
        # tag 即为顶点的标记字典,可以通过顶点查询它的标记
        tag={i : i for i in range(1,n+1)} # 每个点的父节点初始化为自己,key是自己,value是标号。
        result=[] # len(result)=边数=顶点-1
        sum=0
        '遍历每一条边'
        for u,v,w in cost:
            # u,v是顶点,w是边的权重
            '如果两个顶点对应的值不相等(不产生回路)将边加入到最小生成树中' \
            '如果相同(有回路,跳过此边)'
            if (tag[u]!=tag[v]):
                result.append([u,v,w])
                # 把边对应的两个节点对应的编号置为相同,大的向小的看齐。
                max_v = max(tag[u], tag[v]) # 找顶点标号大的
                min_v = min(tag[u], tag[v]) # 找顶点标号小的
                for i in range(1, n+1):  # 找所有的顶点
                    if tag[i] == max_v: # 顶点对应的标号等于大的那个
                        tag[i] = min_v  # 大的标号向小的标号看齐
                sum+=w
                if(len(result)==n-1):
                    return sum
        return sum

全部评论

相关推荐

点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
11-27 10:46
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务