题解 | #最小生成树#

最小生成树

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

全部评论

相关推荐

宇算唯航:目测实缴资本不超100W的小公司
点赞 评论 收藏
分享
Twilight_m...:表格简历有点难绷。说说个人看法: 1.个人基本情况里好多无意义信息,什么婚姻状况、健康状况、兴趣爱好、户口所在地、身份证号码、邮政编码,不知道的以为你填什么申请表呢。 2.校内实践个人认为对找工作几乎没帮助,建议换成和测开有关的项目,实在没得写留着也行。 3.工作经历完全看不出来是干什么的,起码看着和计算机没啥关系,建议加强描述,写点你在工作期间的实际产出、解决了什么问题。 4.个人简述大而空,看着像AI生成,感觉问题最大。“Python,C,C++成为我打造高效稳定服务的得力工具”、“我渴望凭借自身技术知识与创新能力,推动人工智能技术的应用发展,助力社会实现智能化转型”有种小学作文的美感。而且你确定你个人简述里写的你都会嘛?你AI这块写的什么“深入研究”,发几篇顶会的硕博生都不一定敢这么写。而且你AI这块的能力和软测也完全无关啊。个人简述建议写你对哪些技术栈、哪些语言、哪些生产工具的掌握,写的有条理些,而且最好是和测开强相关的。
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
05-29 20:12
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务