Python 经典贪心算法之Prim算法案例详解

第1行：2个数N,M中间用空格分隔，N为点的数量，M为边的数量。（2 <= N <= 1000, 1 <= M <= 50000)
第2 - M + 1行：每行3个数S E W，分别表示M条边的2个顶点及权值。(1 <= S, E <= N，1 <= W <= 10000)

maxv=10001
n,m=list(map(int,input().split()))
E=[]
V=set([1])
cost=[]
for i in range(n+1):
    a=[]
    for j in range(n+1):
        a.append(maxv)
    cost.append(a)
for i in range(m):
    s,e,w=list(map(int,input().split()))
    cost[s][e]=w
    cost[e][s]=w
closet=[0]
lowcost=[maxv]
for i in range(1,n+1):
    closet.append(1)
    lowcost.append(cost[1][i])
ans=0
for i in range(n-1):
    k=0
    for j in range(2,n+1):
        if (lowcost[j]!=0) and (lowcost[j]<lowcost[k]):k=j

    for j in range(2,n+1):
        if cost[j][k]<lowcost[j]:
            lowcost[j]=cost[j][k]
            closet[j]=k
    ans+=lowcost[k]
    lowcost[k]=0
print(ans)