算法 | 狄克斯特拉算法

算法 | 狄克斯特拉算法

1. 简介

简介：

摘自百度百科

迪杰斯特拉算法(Dijkstra)是由荷兰计算机科学家狄克斯特拉于1959年提出的，因此又叫狄克斯特拉算法。是从一个顶点到其余各顶点的最短路径算法，解决的是有权图中最短路径问题。迪杰斯特拉算法主要特点是从起始点开始，采用贪心算法的策略，每次遍历到始点距离最近且未访问过的顶点的邻接节点，直到扩展到终点为止。

算法逻辑：

1. 保存启始点到相邻的权重到权重记录，保存相邻的父节点记录到父节点记录。
2. 查找权重记录中之前没有处理过的最小权重记录，如果不存在结束。
3. 计算启始点到最小权重记录点相邻点权重。如果不存在直接保存到权重记录和父节点记录，如果存在比较是否小于记录值，小于记录值覆盖旧记录。
4. 最后标记当前最小权重记录点为已处理。
5. 一直重复2-4步骤，直到没有未处理的最小权重记录退出。

注意：

1. 不能处理负权重，如果需要请查看贝尔曼-福德算法（Bellman-Fordalgorithm）。

2. 图示

3. 演示

package main

import (
	"fmt"
	"math"
)

func main() {
	graph := make(map[string]map[string]uint32)

	graph["Start"] = map[string]uint32{}
	graph["Start"]["A"] = 6
	graph["Start"]["B"] = 2

	graph["A"] = map[string]uint32{}
	graph["A"]["Dest"] = 1

	graph["B"] = map[string]uint32{}
	graph["B"]["A"] = 3
	graph["B"]["Dest"] = 5

	graph["Dest"] = map[string]uint32{}

	start := "Start"
	Dest := "Dest"
	costs, parents := findShortestPath(graph, start)
	fmt.Println(costs[Dest], parents)
}

func findShortestPath(graph map[string]map[string]uint32, start string) (map[string]uint32, map[string]string) {
	costs := make(map[string]uint32)
	parents := make(map[string]string)
	processed := make(map[string]struct{})

	for node, cost := range graph[start] {
		parents[node] = start
		costs[node] = cost
	}

	for {
		lowestCostNode := findLowestCost(costs, processed)
		if len(lowestCostNode) == 0 {
			break
		}
		for node, cost := range graph[lowestCostNode] {
			oldCost, ok := costs[node]
			newCost := cost + costs[lowestCostNode]
			if !ok || newCost < oldCost {
				// 不存在费用 || 当前费用小于之前费用
				parents[node] = lowestCostNode
				costs[node] = newCost
			}
		}

		// 记录当前点已走过
		processed[lowestCostNode] = struct{}{}
	}

	return costs, parents
}

func findLowestCost(costs map[string]uint32, processed map[string]struct{}) string {
	lowestNode := ""
	var lowestCost uint32 = math.MaxUint32
	for node, cost := range costs {
		if _, ok := processed[node]; ok {
			continue
		}

		if cost < lowestCost {
			lowestNode = node
			lowestCost = cost
		}
	}

	return lowestNode
}

4. 参考

• 《图解算法》
• 迪克斯特拉算法