最小生成树--Prim算法（Python）

最新推荐文章于 2025-06-28 22:21:52 发布

寸草&春晖

最新推荐文章于 2025-06-28 22:21:52 发布

阅读量4.5k

点赞数 7

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/weixin_44193909/article/details/88894487

1、Prim算法原理
Prim算法在找当前最近顶点时使用到了贪婪算法。Prim算法从任意一个顶点开始，每次选择一个与当前顶点集最近的一个顶点，并将两顶点之间的边加入到树中,直到当前顶点集顶点个数为n或者树中边数为n-1。
2、实战演练
下面给出一个无向图G=（V，E），我们使用Prim算法来找它的最小生成树。
第一步，我们先随意选择一个顶点作为起始点（起始点的选取不会影响最小生成树结果），在此我们一般选择v1作为起始点，现在我们设当前所找到最小生成树里面的顶点为集合V’，所找到的边为集合E’。初始状态如下：V’={v1}； E’={}。
在这里插入图片描述

第二步，查找顶点在V’ ={v1}，V-V’ 集合中的最小权值，如下图，在红线相交的线上找最小值。通过图中我们可以看到边（v1，v8）的权值最小为2，那么将v8加入到V’，边（v1，v8）加入到 E’。此时状态如下：V’={v1，v8}； E’={（v1，v8）}。
在这里插入图片描述
第三步，查找顶点在V’={v1，v8}，V-V’ 集合中的最小权值，如下图，在红线相交的线上找最小值。通过图中我们可以看到边（v8，v9）的权值最小为4，那么将v9加入到V’，边（v8，v9）加入到 E’。此时状态如下：V’ ={v1，v8，v9}； E’={（v1，v8），（v8，v9）}。
在这里插入图片描述
第四步，查找顶点在V’ ={v1，v8，v9}，V-V’ 集合中的最小权值，如下图，在红线相交的线上找最小值。通过图中我们可以看到边（v9，v2）的权值最小为1，那么将v2加入到V’，边（v9，v2）加入到 E’。此时状态如下：V’={v1，v8，v9，v2}； E’={（v1，v8），（v8，v9），（v9，v2）}。
在这里插入图片描述
以此类推，直到找到所有顶点为止，最后得到无向图G的最小生成树MST=（V’，E’），如下图所示：

3、Python实现

#Prim algorithm(python)
from collections import defaultdict
from heapq import heapify, heappush, heappop
import time
start = time.perf_counter()
def Prim(nodes, edges):
    '''  基于最小堆实现的Prim算法  '''
    element = defaultdict(list