func getMinDistanceAndUnselectedNode(distanceMap map[*Node]int, selectedMap map[*Node]void) *Node {
var minNode *Node
minDistance := math.MaxInt64
for node, distance := range distanceMap {
if _, exist := selectedMap[node]; !exist && distance < minDistance {
minNode = node
}
}
return minNode
}
func dijkstra(head *Node) map[*Node]int {
distanceMap := make(map[*Node]int, 32)
distanceMap[head] = 0
selectedMap := make(map[*Node]void, 32)
minNode := getMinDistanceAndUnselectedNode(distanceMap, selectedMap)
for minNode != nil {
distance := distanceMap[minNode]
for _, edge := range minNode.edges {
toNode := edge.to
if _, exit := distanceMap[toNode]; !exit {
distanceMap[toNode] = distance + edge.weitht
} else {
distanceMap[toNode] = int(math.Min(float64(distanceMap[toNode]), float64(distance+edge.weitht)))
}
}
selectedMap[minNode] = void{}
minNode = getMinDistanceAndUnselectedNode(distanceMap, selectedMap)
}
return distanceMap
}
func main() {
arr := [][]int{{1, 2, 1}, {2, 1, 1}, {1, 3, 2}, {3, 1, 2}, {2, 3, 3}, {3, 2, 3}}
graph := NewGraph(arr)
ret := kruskalMST(graph)
fmt.Println(ret)
}
dijkstra
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
- 文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
