我最近学习了图论,我正在尝试用一个权矩阵来实现Kruskal算法来寻找图中的最小生成树,我得到了一个矩阵的正确输出和另一个矩阵的出界误差! 它给了我一个错误:
[[1000,16,12,21,1000,1000,1000],[16,1000,1000,17,20,1000,1000],[12,1000,1000,28,1000,31,1000],[21,17,28,1000,18,19,23],[1000,20,1000,18,1000,1000,11],[1000,1000,31,19,1000,1000,27],[1000,1000,1000,23,11,27,1000]]
我得到正确答案的矩阵如下: (注:1000表示无穷大的重量
vertices=5
spset=[True]*5
wt=[[1000,1,3,4,1000],[1,1000,5,1000,7],[3,5,1000,6,8],[4,1000,6,1000,2],[1000,7,8,2,1000]]
row=[0]
for i in xrange(vertices-1):
row_num,col_num,min_no=-1,-1,1000
for i in row:
temp=min(wt[row[i]])
if(min_no>temp):
min_no=temp
row_num=i
col_num=wt[i].index(temp)
print str(min_no)+"("+str(row_num)+","+str(col_num)+")"
spset[col_num]=False
wt[col_num][row_num]=1000
for i in xrange(vertices):
wt[i][col_num]=1000
row.append(col_num)
d=raw_input()
在下面的代码块中
i
正在对row
的元素进行迭代。如果row
有两个元素[0,2]
,那么当i
变为2时,row[i]
将给出IndexError: list index out of range
。如果要遍历
row
的元素,请改用for i in range(len(row))
。相关问题 更多 >
编程相关推荐