Python中的克鲁斯卡尔算法答案

【问题标题】：Kruskal's Algorithm in PythonPython中的克鲁斯卡尔算法
【发布时间】：2014-01-28 10:54:50
【问题描述】：

我最近学习了图论，我正在尝试实现 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()

【问题讨论】：

提示：避免代码中的空格并不能让它运行得更快。它只是让阅读变得更加困难:-)

标签： python algorithm graph-theory kruskals-algorithm

【解决方案1】：

在下面的代码块中，

for i in row:
    temp=min(wt[row[i]])

i 正在迭代 row 的元素。如果row 有两个元素[0,2]，那么当i 变为2 时，row[i] 将给出IndexError: list index out of range。

如果您想遍历row 的元素，请改用for i in range(len(row))。

【讨论】：