题解 | #最小生成树#
最小生成树
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