Python图论算法(2)——prim算法

Python图论算法(二)——prim算法

算法的思想就不多说了。主要对代码解释一下。

graph = [[0,4,3,2],
         [4,0,1,4],
         [3,1,0,2],
         [2,4,2,0]]
n = 4
flags = [True,True,True,True]
queue = []
#flags是存储节点访问情况的,true为待访问节点
#queue存储节点访问顺序

queue.append(0)
flags[0] = False
<pre name="code" class="python">#现将初始节点加入queue
while(len(queue) != n): minedge = 10000 for i in range(len(queue)): for j in range(n): if(flags[j] and minedge > graph[queue[i]][j]): minedge = graph[queue[i]][j] minj = j print minedge
#这一段就是在queue里的节点的邻居里需找安全边
 flags[minj] = False queue.append(minj) print queue


输入的结果就是选择边的权重和节点扩展的顺序。