77 lines
1.4 KiB
Go
77 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
h "git.bullercodeworks.com/brian/adventofcode/helpers"
|
|
)
|
|
|
|
func main() {
|
|
inp := h.StdinToCoordMap()
|
|
part1(inp)
|
|
fmt.Println()
|
|
part2(inp)
|
|
}
|
|
|
|
func part1(inp h.CoordByteMap) {
|
|
var total int
|
|
heads := inp.FindAll('0')
|
|
for _, h := range heads {
|
|
trails := findTrails(inp, h, h, false)
|
|
total = total + len(trails)
|
|
}
|
|
fmt.Println("# Part 1")
|
|
fmt.Println("Scores:", total)
|
|
}
|
|
|
|
func part2(inp h.CoordByteMap) {
|
|
var total int
|
|
heads := inp.FindAll('0')
|
|
for _, h := range heads {
|
|
trails := findTrails(inp, h, h, true)
|
|
total = total + len(trails)
|
|
}
|
|
fmt.Println("# Part 1")
|
|
fmt.Println("Scores:", total)
|
|
}
|
|
|
|
func findTrails(m h.CoordByteMap, st h.Coordinate, curr h.Coordinate, unique bool) []h.Coordinate {
|
|
currBt := m.Get(curr)
|
|
if currBt == '9' {
|
|
return []h.Coordinate{curr}
|
|
}
|
|
|
|
var trails []h.Coordinate
|
|
for _, nx := range []h.Coordinate{curr.North(), curr.East(), curr.South(), curr.West()} {
|
|
if m.ContainsCoord(nx) && m.Get(nx) == currBt+1 {
|
|
res := findTrails(m, st, nx, unique)
|
|
if unique {
|
|
trails = append(trails, res...)
|
|
} else {
|
|
trails = appendUnique(trails, res...)
|
|
}
|
|
}
|
|
}
|
|
return trails
|
|
}
|
|
|
|
func appendUnique(sl []h.Coordinate, n ...h.Coordinate) []h.Coordinate {
|
|
if len(n) == 0 {
|
|
return sl
|
|
}
|
|
if len(n) > 1 {
|
|
for j := range n {
|
|
sl = appendUnique(sl, n[j])
|
|
}
|
|
return sl
|
|
}
|
|
|
|
for i := range sl {
|
|
if sl[i].Equals(n[0]) {
|
|
return sl
|
|
}
|
|
}
|
|
sl = append(sl, n[0])
|
|
return sl
|
|
}
|