我在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。