Compare commits
No commits in common. "e40c41ddfb5adfdd5bd12bee0829b479c7524264" and "3cccdc7eb80d1dea7fc51501fa6d74684f1f1bc2" have entirely different histories.
e40c41ddfb
...
3cccdc7eb8
@ -1,66 +0,0 @@
|
||||
package solver
|
||||
|
||||
import (
|
||||
"maze-solver/maze"
|
||||
"maze-solver/utils"
|
||||
"sort"
|
||||
)
|
||||
|
||||
type AStarSolver struct {
|
||||
dist_from_start map[*maze.Node]int
|
||||
dist_from_end map[*maze.Node]int
|
||||
parent map[*maze.Node]*maze.Node
|
||||
stack sorted_stack
|
||||
}
|
||||
|
||||
func (s *AStarSolver) Solve(m *maze.Maze) *maze.SolvedMaze {
|
||||
defer utils.Timer("A* algorithm", 2)()
|
||||
s.dist_from_start = make(map[*maze.Node]int, len(m.Nodes))
|
||||
s.dist_from_end = make(map[*maze.Node]int, len(m.Nodes))
|
||||
s.parent = make(map[*maze.Node]*maze.Node, len(m.Nodes))
|
||||
|
||||
current, end := m.Nodes[0], m.Nodes[len(m.Nodes)-1]
|
||||
|
||||
for _, node := range m.Nodes {
|
||||
s.dist_from_start[node] = 0
|
||||
s.dist_from_end[node] = int(node.Coords.Distance(end.Coords))
|
||||
}
|
||||
|
||||
for current != end {
|
||||
current.Visited = true
|
||||
|
||||
for _, child := range []*maze.Node{current.Left, current.Right, current.Up, current.Down} {
|
||||
if child != nil {
|
||||
dist := s.dist_from_start[current] + int(current.Coords.Distance(child.Coords))
|
||||
if !child.Visited {
|
||||
s.parent[child] = current
|
||||
s.dist_from_start[child] = dist
|
||||
s.stack.insert(child, &s.dist_from_end)
|
||||
} else if s.dist_from_start[child] > dist {
|
||||
s.parent[child] = current
|
||||
s.dist_from_start[child] = dist
|
||||
sort.Slice(s.stack, func(i, j int) bool {
|
||||
return s.dist_from_end[s.stack[i]] < s.dist_from_end[s.stack[j]]
|
||||
})
|
||||
}
|
||||
}
|
||||
}
|
||||
current = s.stack.pop()
|
||||
}
|
||||
|
||||
solution := make([]*maze.Node, 0, len(m.Nodes))
|
||||
for current != m.Nodes[0] {
|
||||
solution = append(solution, current)
|
||||
current = s.parent[current]
|
||||
}
|
||||
solution = append(solution, m.Nodes[0])
|
||||
|
||||
for i, j := 0, len(solution)-1; i < j; i, j = i+1, j-1 {
|
||||
solution[i], solution[j] = solution[j], solution[i]
|
||||
}
|
||||
|
||||
return &maze.SolvedMaze{
|
||||
Maze: m,
|
||||
Solution: solution,
|
||||
}
|
||||
}
|
@ -3,6 +3,7 @@ package solver
|
||||
import (
|
||||
"maze-solver/maze"
|
||||
"maze-solver/utils"
|
||||
"slices"
|
||||
"sort"
|
||||
)
|
||||
|
||||
@ -12,6 +13,8 @@ type DijkstraSolver struct {
|
||||
stack sorted_stack
|
||||
}
|
||||
|
||||
type sorted_stack []*maze.Node
|
||||
|
||||
func (s *DijkstraSolver) Solve(m *maze.Maze) *maze.SolvedMaze {
|
||||
defer utils.Timer("Dijkstra algorithm", 2)()
|
||||
s.dist_from_start = make(map[*maze.Node]int, len(m.Nodes))
|
||||
@ -61,3 +64,22 @@ func (s *DijkstraSolver) Solve(m *maze.Maze) *maze.SolvedMaze {
|
||||
Solution: solution,
|
||||
}
|
||||
}
|
||||
|
||||
func (s *sorted_stack) insert(node *maze.Node, dists *map[*maze.Node]int) {
|
||||
var dummy *maze.Node
|
||||
*s = append(*s, dummy) // extend the slice
|
||||
|
||||
i, _ := slices.BinarySearchFunc(*s, node, func(e, t *maze.Node) int {
|
||||
return (*dists)[t] - (*dists)[e]
|
||||
})
|
||||
|
||||
copy((*s)[i+1:], (*s)[i:]) // make room
|
||||
(*s)[i] = node
|
||||
}
|
||||
|
||||
func (s *sorted_stack) pop() *maze.Node {
|
||||
last_i := len(*s) - 1
|
||||
ret := (*s)[last_i]
|
||||
*s = (*s)[:last_i]
|
||||
return ret
|
||||
}
|
||||
|
@ -17,14 +17,12 @@ const (
|
||||
_DFS = "dfs"
|
||||
_BFS = "bfs"
|
||||
_Dijkstra = "dijkstra"
|
||||
_AStar = "a-star"
|
||||
)
|
||||
|
||||
var TYPES = []string{
|
||||
_DFS,
|
||||
_BFS,
|
||||
_Dijkstra,
|
||||
_AStar,
|
||||
}
|
||||
|
||||
func (f *SolverFactory) Get() Solver {
|
||||
@ -33,8 +31,6 @@ func (f *SolverFactory) Get() Solver {
|
||||
return &DFSSolver{}
|
||||
case _BFS:
|
||||
return &BFSSolver{}
|
||||
case _AStar:
|
||||
return &AStarSolver{}
|
||||
case _Dijkstra:
|
||||
return &DijkstraSolver{}
|
||||
}
|
||||
|
@ -1,27 +0,0 @@
|
||||
package solver
|
||||
|
||||
import (
|
||||
"maze-solver/maze"
|
||||
"slices"
|
||||
)
|
||||
|
||||
type sorted_stack []*maze.Node
|
||||
|
||||
func (s *sorted_stack) insert(node *maze.Node, weights *map[*maze.Node]int) {
|
||||
var dummy *maze.Node
|
||||
*s = append(*s, dummy) // extend the slice
|
||||
|
||||
i, _ := slices.BinarySearchFunc(*s, node, func(e, t *maze.Node) int {
|
||||
return (*weights)[t] - (*weights)[e]
|
||||
})
|
||||
|
||||
copy((*s)[i+1:], (*s)[i:]) // make room
|
||||
(*s)[i] = node
|
||||
}
|
||||
|
||||
func (s *sorted_stack) pop() *maze.Node {
|
||||
last_i := len(*s) - 1
|
||||
ret := (*s)[last_i]
|
||||
*s = (*s)[:last_i]
|
||||
return ret
|
||||
}
|
Loading…
Reference in New Issue
Block a user