OneCompiler

prim

136

inf=9999
v=6
g=[[0,3,0,0,6,5],
[3,0,1,0,0,4],
[0,1,0,6,0,4],
[0,0,6,0,8,5],
[6,0,0,8,0,2],
[5,4,4,5,2,0]]
sel=[0,0,0,0,0,0]
no_edge=0
sel[0]=True
print("Edge : Weight\n")
while(no_edge<v-1):
mini=inf
x=0
y=0
for i in range(v):
if sel[i]:
for j in range(v):
if((not sel[j]) and g[i][j]):
if mini>g[i][j]:
mini=g[i][j]
x=i
y=j
print(str(x)+"-"+str(y)+":"+str(g[x][y]))
sel[y]=True
no_edge=no_edge+1