106 lines
1.8 KiB
Go
106 lines
1.8 KiB
Go
|
package day_06
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Direction int
|
||
|
|
|
||
|
|
const (
|
||
|
|
UP Direction = iota
|
||
|
|
RIGHT Direction = iota
|
||
|
|
DOWN Direction = iota
|
||
|
|
LEFT Direction = iota
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
VISITED = 'X'
|
||
|
|
OCCUPIED = '#'
|
||
|
|
)
|
||
|
|
|
||
|
|
var directions = map[rune]Direction{
|
||
|
|
'^': UP,
|
||
|
|
'>': RIGHT,
|
||
|
|
'v': DOWN,
|
||
|
|
'<': LEFT,
|
||
|
|
}
|
||
|
|
|
||
|
|
// task:https://adventofcode.com/2024/day/5
|
||
|
|
// short summary - count visited steps
|
||
|
|
func SolveBasic(input string) int {
|
||
|
|
grid := loadGrid(input)
|
||
|
|
x, y, dir := findPlayer(grid)
|
||
|
|
|
||
|
|
for inGrid(grid, x, y) {
|
||
|
|
grid[y][x] = VISITED
|
||
|
|
x, y, dir = nextSpot(grid, x, y, dir)
|
||
|
|
}
|
||
|
|
|
||
|
|
sum := 0
|
||
|
|
for _, row := range grid {
|
||
|
|
for _, value := range row {
|
||
|
|
if value == VISITED {
|
||
|
|
sum++
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return sum
|
||
|
|
}
|
||
|
|
|
||
|
|
func nextSpot(grid [][]rune, x int, y int, dir Direction) (int, int, Direction) {
|
||
|
|
for {
|
||
|
|
nx, ny := calculateNext(x, y, dir)
|
||
|
|
if !inGrid(grid, nx, ny) {
|
||
|
|
return nx, ny, dir
|
||
|
|
} else if grid[ny][nx] != OCCUPIED {
|
||
|
|
return nx, ny, dir
|
||
|
|
} else {
|
||
|
|
dir = (dir + 1) % 4
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func calculateNext(x int, y int, dir Direction) (int, int) {
|
||
|
|
switch dir {
|
||
|
|
case UP:
|
||
|
|
return x, y - 1
|
||
|
|
case RIGHT:
|
||
|
|
return x + 1, y
|
||
|
|
case DOWN:
|
||
|
|
return x, y + 1
|
||
|
|
case LEFT:
|
||
|
|
return x - 1, y
|
||
|
|
}
|
||
|
|
panic("unknown direction")
|
||
|
|
}
|
||
|
|
|
||
|
|
func inGrid(grid [][]rune, x int, y int) bool {
|
||
|
|
return x >= 0 && y >= 0 && x < len(grid[0]) && y < len(grid)
|
||
|
|
}
|
||
|
|
|
||
|
|
func findPlayer(grid [][]rune) (int, int, Direction) {
|
||
|
|
for y, row := range grid {
|
||
|
|
for x, value := range row {
|
||
|
|
if dir, present := directions[value]; present {
|
||
|
|
return x, y, dir
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
panic("Couldn't find player")
|
||
|
|
}
|
||
|
|
|
||
|
|
func loadGrid(input string) [][]rune {
|
||
|
|
substrings := strings.Split(input, "\n")
|
||
|
|
var grid [][]rune
|
||
|
|
for _, line := range substrings {
|
||
|
|
grid = append(grid, []rune(line))
|
||
|
|
}
|
||
|
|
return grid
|
||
|
|
}
|
||
|
|
|
||
|
|
// task:https://adventofcode.com/2024/day/5#part2
|
||
|
|
// short summary - fix and count invalid columns
|
||
|
|
func SolveComplex(input string) int {
|
||
|
|
return 0
|
||
|
|
}
|