package/gonum
1
0
Fork 0
mirror of https://github.com/gonum/gonum.git synced 2025-10-14 11:23:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Updated Dijkstra

Browse Source
This commit is contained in:
Jsor
2014-02-06 03:01:15 -07:00
parent a5c25df8d6
commit 8c477930f7
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
search/graphSearch.go
View File

@@ -89,8 +89,8 @@ func BreadthFirstSearch(start, goal gr.Node, graph gr.Graph) ([]gr.Node, int) {
// Like A*, Dijkstra's Algorithm likely won't run correctly with negative edge weights -- use Bellman-Ford for that instead // Like A*, Dijkstra's Algorithm likely won't run correctly with negative edge weights -- use Bellman-Ford for that instead
// //
// Dijkstra's algorithm usually only returns a cost map, however, since the data is available this version will also reconstruct the path to every node // Dijkstra's algorithm usually only returns a cost map, however, since the data is available this version will also reconstruct the path to every node
func Dijkstra(source gr.Node, graph gr.Graph, Cost func(gr.Node, gr.Node) float64) (paths map[int][]gr.Node, costs map[int]float64) { func Dijkstra(source gr.Node, graph gr.Graph, cost gr.CostFun) (paths map[int][]gr.Node, costs map[int]float64) {
successors, _, _, _, _, _, Cost, _ := setupFuncs(graph, Cost, nil) successors, _, _, _, _, _, cost, _ := setupFuncs(graph, cost, nil)
nodes := graph.NodeList() nodes := graph.NodeList()
openSet := &aStarPriorityQueue{nodes: make([]internalNode, 0), indexList: make(map[int]int)} openSet := &aStarPriorityQueue{nodes: make([]internalNode, 0), indexList: make(map[int]int)}
@@ -111,7 +111,7 @@ func Dijkstra(source gr.Node, graph gr.Graph, Cost func(gr.Node, gr.Node) float6
closedSet.Add(node.ID()) closedSet.Add(node.ID())
for _, neighbor := range successors(node) { for _, neighbor := range successors(node) {
tmpCost := costs[node.ID()] + Cost(node, neighbor) tmpCost := costs[node.ID()] + cost(node, neighbor)
if cost, ok := costs[neighbor.ID()]; !ok { if cost, ok := costs[neighbor.ID()]; !ok {
costs[neighbor.ID()] = tmpCost costs[neighbor.ID()] = tmpCost
predecessor[neighbor.ID()] = node predecessor[neighbor.ID()] = node