为什么此代码通过Prim的算法使最小生成树错误?

时间:2019-06-23 19:27:55

标签: algorithm go queue priority-queue

我在Golang上有一些Prim的算法代码。 任务是找到边缘长度最小的最小生成树。但是,如果顶点数大于3,则答案错误。

所以我试图重写Prim的算法,但是没有任何改变。

func MST_Prim(vertices []*Vertex, vertNum int, edgeNum int) []*Edge {
    treeEdges := make([]*Edge, 0)

    for i := 0; i < vertNum; i++ {
        vertices[i].index = -1
    }

    queue := make(PriorityQueue, 0)
    v := vertices[0]
    for i := 0; i < vertNum; i++ {
        v.index = -2

        for _, edge := range vertices[v.id].edges {
            u := vertices[edge.v]

            if u.index == -1 {
                u.priority = edge.len
                u.value = v.id
                queue.Push(u)
            } else if u.index != -2 && edge.len < u.priority {
                u.value = v.id
                queue.update(u, v.value, edge.len)
            }
        }

        if queue.Len() == 0 {
            break
        }

        v = queue.Pop().(*Vertex)
        treeEdges = append(treeEdges, &Edge{v.id, v.value, v.priority})
    }

    return treeEdges
}

我需要在最小生成树中打印最小的边长之和。 输入:

4
4
1 2 462
0 3 343
0 1 430
1 3 146

我得到1038作为答案,但正确答案是951。

0 个答案:

没有答案
相关问题