Prim演算法最小生成樹 Python

2021-10-07 08:27:21 字數 2188 閱讀 6655

qwurey的文章寫的很好懂了,但是自己**再寫出來確實不一樣,主要還是練一下基本功,自己基本功還不夠,**勉強湊合吧,可能過多一陣自己回來再看就能寫出更好的。作者的圖很好看,用什麼畫的呢。

# 表示已經加入mst

disconnect =

10000

# 表示無通路

defprim

(graph:

dict

, start:

str)

->

list

: lowcost =

[disconnect]

*len

(graph)

lowcost[

int(start)-1

]= visited

mst =

[start]

*len

(graph)

# 記錄頂點的歷史更新,用於回溯

mst[

int(start)-1

]=0 visted =

vist =

# 用於輸出

node = start

# 更新lowcost和mst

while

any(mst)

: lowcost, mst = update(graph[node]

, node, lowcost, mst)

if node == mst[lowcost.index(

min(lowcost))]

:'->'

+str

(lowcost.index(

min(lowcost))+

1)) node, mst[lowcost.index(

min(lowcost))]

=str

(lowcost.index(

min(lowcost))+

1),0

# 找出到開始的頂點權值最小的頂點

else

:str

(mst[lowcost.index(

min(lowcost))]

)+'->'

+str

(lowcost.index(

min(lowcost))+

1)) node, mst[lowcost.index(

min(lowcost))]

=str

(lowcost.index(

min(lowcost))+

1),0

lowcost[lowcost.index(

min(lowcost))]

= visited

return vist

defupdate

(edge:

list

, thisnode:

str, lowcost:

list

, mst:

list):

for i in edge:

if lowcost[

int(i[0]

)-1]

!= visited and i[1]

< lowcost[

int(i[0]

)-1]

: lowcost[

int(i[0]

)-1]

= i[1]

mst[

int(i[0]

)-1]

= thisnode

return lowcost, mst

if __name__ ==

"__main__"

: g =

v = prim(g,

'1')

for i in v:

print

(i)

connected to pydev debugger (build 191.7141.48)

1->3

3->6

6->4

3->2

2->5

最小生成樹(prim演算法)

最小生成樹是資料結構中圖的一種重要應用,它的要求是從乙個帶權無向完全圖中選擇n 1條邊並使這個圖仍然連通 也即得到了一棵生成樹 同時還要考慮使樹的權最小。prim演算法要點 設圖g v,e 其生成樹的頂點集合為u。把v0放入u。在所有u u,v v u的邊 u,v e中找一條最小權值的邊,加入生成樹...

最小生成樹 Prim演算法

prim 演算法 以領接矩陣儲存 圖g bool b i 表示頂點i是否被訪問,初始化時候memset b,false,sizeof b b 0 value,表示從第0個節點開始。用value i 表示節點i到最小生成樹a中定點的最小距離。例如value 1 a 0 1 int sum記錄權值和 i...

最小生成樹 prim 演算法

一 演算法描述 假設存在連通帶權圖g v,e 其中最小生成樹為t,首先從圖中隨意選擇一點s屬於v作為起始點,並將其標記後加入集合u 中。然後演算法重複執行操作為在所有v屬於u,u屬於v u的邊 v0,u0 屬於e中找一條代價最小的邊並加入集合t,同時將u0併入u,直到u v為止。這是,t中必有n 1...