使用具有优先级队列的Prim算法?

时间:2014-02-28 05:24:07

标签: python algorithm priority-queue prims-algorithm

我正在使用Prim的算法和优先级队列来解决图形。我得到了一个包含处理边缘和图形本身的代码的类。我还需要使用优先级队列来查找最小生成树。

你们如何解决这样的问题?你会获得每个节点的所有边缘,将它放入优先级队列然后从那里开始吗?

1 个答案:

答案 0 :(得分:0)

对于Prims算法,我的优先级队列有pairs (weight, vertex),按权重排序。 这是伪代码。请注意,如果它优先于您为该边缘遇到的所有边缘,则只需将边缘添加到顶点的优先级队列。

key[v] = Infinity for all vertex v
vis[v] = false for all vertex v
Q.Push(0, 1) // weight 0, and 1 is a valid vertex
while (Q is not empty) :
  curV = first element of Q
  curW = second element of Q
  Q.pop
  if (vis[curV]) continue
  add curV to spanning tree
  for u in adjacency list of curV :
    if (not vis[u] and key[u] > weight(u, v)):
      Q.push(u, weight(u,v)
      key[u] = weight(u,v)