Prim算法的Python实现

话不投机半句多。
代码一一如下。

# Minimum spanning tree. Prim algorithm
import sys

# 同样的，这里也是为了引入无穷大

graphMatrix = [[0, 54, 32, 7, 50, 60], [54, 0, 21, 58, 76, 69], [32, 21, 0, 35, 67, 66],
               [7, 58, 35, 0, 50, 62], [50, 76, 67, 50, 0, 14], [60, 69, 66, 62, 14, 0]]

treeDis = graphMatrix[0]  # 各个点距离生成树的最短距离列表
visited = [0 for i in range(6)]  # 已经访问过的节点将被置为1
visited[0] = 1
# 不在树中的点距离树有最短距离，在树中对应的距离最小的那个店
# 比如neighbor[1]=0表示在节点1还不在树中时，它离树中的节点0距离最小
neighbor =