Python圖論演算法（二）——prim演算法

標籤：python   演算法   prim   

演算法的思想就不多說了。主要對代碼解釋一下。

graph = [[0,4,3,2],         [4,0,1,4],         [3,1,0,2],         [2,4,2,0]]n = 4flags = [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

輸入的結果就是選擇邊的權重和節點擴充的順序。

Python圖論演算法（二）——prim演算法