From 18bf5a4aef46bfeac54aa5256e3b40d42e0a37ba Mon Sep 17 00:00:00 2001 From: xuu Date: Sun, 17 Dec 2023 09:43:04 -0700 Subject: [PATCH] chore: add day16 --- day16/example.txt | 10 +++ day16/input.txt | 110 ++++++++++++++++++++++++++ day16/main.go | 189 +++++++++++++++++++++++++++++++++++++++++++++ day16/main_test.go | 50 ++++++++++++ tools_test.go | 80 +++++++++++++++++++ 5 files changed, 439 insertions(+) create mode 100644 day16/example.txt create mode 100644 day16/input.txt create mode 100644 day16/main.go create mode 100644 day16/main_test.go create mode 100644 tools_test.go diff --git a/day16/example.txt b/day16/example.txt new file mode 100644 index 0000000..c78b2e7 --- /dev/null +++ b/day16/example.txt @@ -0,0 +1,10 @@ +.|...\.... +|.-.\..... +.....|-... +........|. +.......... +.........\ +..../.\\.. +.-.-/..|.. +.|....-|.\ +..//.|.... \ No newline at end of file diff --git a/day16/input.txt b/day16/input.txt new file mode 100644 index 0000000..d421735 --- /dev/null +++ b/day16/input.txt @@ -0,0 +1,110 @@ +\.......................\..........................................-..................-.../................-.. +....../.......\...-......-.|...\..............\.................|............././....-........................ +...............-...-........................................\......\..-.............|......-....\....-........ +..\..|.|...........................-.|........................................./.|....|................|...... +|../................|............................................................\........../..\.............. +........\............../................/..........................\...\....|............./....-.........\\... +.................-...-...................\../........-|.................../......-...........|................ +........-............/...|-............-.....\.|..............-....-../\../..................\...\././........ +........................-|................................................................-........-.......... +-......./.................|.....\.................\.....//............-........./..../....|................... +....-.............-........-..........-|.......|...../...\...................\.....................|.......... +...|...\.........|.................|.....|....-....\.......\./........................-.../..../.....-........ +....|....-.-..............\.........|.......................-.......-..............\................/......... +.....|.\...........-.................-.-........../...\............|...............|...|......./.......|...../ +.....\.........................-.../..-./...\...................../......\...-.......|.\........-|............ +............|.....................................-....../....\.............../-...|...\..\................./. +......-./|....../............|.........\............-../...-......./.\..\....../......|/...\.................. +................................|.......|..-..../.............................../.-..................-..-/..|. +...............................-......./......................|./...............-............./....../........ +..................../....................\........\...........................-............/.../.............. +.|||......../.................|.................../..\............................-......./..|...|/.-......... +......-......./\....../.........|....../.......-...............|.\........-............/.../.................. +.....................-.\/|................//.../.........|.|......\.-..|.........|.........\.................. +.........................\/.......|.|............................|........|.............................././.. +.................................|.................\../..|/.................\.................-...-........-/. +......................-...-.........../...........................\....\......-..-.......\.-....\....|........ +......-......../..............\/..................\..|........-..............|...-...../......|............... +|.......|.../....|..............//.....................|..........|/......../................|................ +...|.|..................../|..../.-.......-...........|.....|./....\......-......-................|........... +......\/./.|\..../|.|..|..........-....|...........|.-.|...................-.\.....|....................\..... +......../.\......................|.....|...........|.........../.......|....|...|.......-......../......\...-. +...|...-...||......-..|..-...........|.-.................|.................................-.................. +..............-.\.-........|......\..........................................\./.\........../.....|........... +../\.........|...............|............|.\/.-/.................-....|...-.................-.....|..../..... +....................../......../..../-.....\........................\..-.\....|....-.........../-............. +.................................|................|.........|................../|.....|....\\............/.\.. +...///./.....\.................|/.....................\...................-............|................|..... +............-...........................--........\................/.............-\................\.....-.../ +...............\.........../|............................|.....................\...-...\.....................- +...\.............|.............|................\............\|...../...|.........................|........... +..................|......|..........-.-..............\.........................\.............................. +....-................../.........../...|................/...............|./.|.../....\..../......./......//... +...../....../..............................|...\../..........-..........-........-..//..../................... +...............-......|......................|./........./............/..-...........|../.....-........-./.... +..\.........................|...-.......\.................|..............-........\....-.............-....|... +.......|......|....../........\...........|......-........../......-..................../....................\ +.......................|....-..|........./.|.-...............|..\......\.....-.......\...-/....|/.........\... +\......\..-.....................\..........................\................................../...........\... +-....-......../............................................................../.\.......|..............|....-.. +........|......................|.............-....\................................./..\.......|..........-... +...-...............|........./......\...../.......|...|...............\.../...\.../......|.....-.\......\..... +............./.....|......................|.....................................\...||.................-....|. +./.....-................/........|./....-......../......./..............\..................................\.. +.../..../\........-..........................|.-/.|..\................\\-.......|......./...||.......|........ +.....-.........................-...............\...........|.............-.\.../.|......-.|......../...-.....- +......\....................|...../......................|.../......-..|............|......./.../.........|.... +...|...........-....................................../.|............./....|..../........./.....\../.......... +....../.../................\........................\...\....\......\............../|...............\.-....... +./....\.../.............|............................-.../............../............/...........\...|........ +............................|.................|/..........//.........../..-...............|................... +.....\......\.................|.....|...\-./...............\................|....../.....-..\........-....-... +........................./-........\................./........-/.|....-.......\............-/\......|.\...-... +..../../..|.|...............\.......................................-......................................... +...........|...................................................-...-......|..................\............../. +/....|......................................................................|\.../.......-...........-...|.... +......|.-.\......-........|.....-..|.\..............-......./......-|.........................-............... +............\......................................|........................|.-............................... +.........-.........../....\./....../.-...-..//.........../.......\\........................................... +........\......../............./...-......\.../..|..|............................../......../..\.............. +--.........|.....-...|............../...|.......|..............\|............|.........................|.-.... +-\......./..............-....-....\...|.-........\/.\./............-/...\........|-.........../../...\........ +.\|\..............-......-.\...-....\..........................\........|...........-......./.............../. +..........................|...\..../.../..........\.............-...//...............\..................-..... +........./..........................\............|-.....|..-...|......../........-...........................\ +.......\.../.....\....../.......|..|...............-......................|....-...................-.|........ +......|.......\.../..............................-..............................|......-.-...................\ +/.....|.......-............/|.....|-......./..../......-........../..|......../.-..\..../....../...|.......... +............/..--..........-............\.......\................\........................../../....|......... +..|..-................|.|..|................/.......\............-//.......................\..\............... +.......................-......-..................|....\................-......\.|....\....|..../../|.-........ +........................-..................-...........................-..-................-................|. +/.\....|.-.................................\..-.....\./.........../-.|..............\..\.....--|/....\........ +.......-..-../..................\..--|..............|...............|\........|...--................|./.\..... +..........|................-.......|..................................................|...................|... +-.-..|....-.|........-./...|.-.........\.....\/........-........\........../................./.../........\... +..................../...|/......................................../............\.|./../................/\|.... +...//.........................\....../..............\../.......-..\....|...................................... +............................./..|./../.........-/..\.......................................................... +......\.....|..................|..................\.....\.-..\...../............................../........... +..........\...........-..........-.....-....\/................../.................................\.....|..\.. +..../.\...................................\............\...\....|.................|.....................-..... +...........-...........................\.......|-../.|....|.....|......................-...../................ +.......|.........\..............|....\...-......../...........-.....-..-...........-..................\...\.|. +...........|.......................|....-..........\............................\.................../.\....... +..../.................\.......-..|.................../........................|...............\...........-.|. +............./..|.-\..../.....-............-....|..\......|......../............-.........|................\.. +...................\...\/......\............|......................\....|...................-./-.............. +..-\...........|.................../.....|....|............................-..|............\.............\.... +.|.....\....\........|..................\...............|..-/.........|.-/...........-............-........... +.....\...../......-............|........./..|............-........................\....-...................... +......\......../........|..............\...-..-...|.......................................................|... +...................|.................................\.....\...........................|../.|.-......-........ +..................|..............\.....\............|........../....../...............\.........\............. +..|...........-..../....\.........--...\................../.............-...-....................-..../....|.. +.|...\...................|......-...........-\..........\.....|.....|..........|.........|....|-../........... +.............\|...................../\...............\....../.......-....\.....................-.............. +.......//.....................\....\......\......\.........|...................../......../.........\......-.. +.\./.........-...-./-.\.............../......|................................\|....\...../....-.............. +.......|......../......../......./.........../...............|............................../..|......../..... +............./.........|.....-./.........../....-......./|......-................|............................ diff --git a/day16/main.go b/day16/main.go new file mode 100644 index 0000000..a955335 --- /dev/null +++ b/day16/main.go @@ -0,0 +1,189 @@ +package main + +import ( + "bufio" + _ "embed" + "fmt" + + aoc "go.sour.is/advent-of-code" +) + +// var log = aoc.Log + +func main() { aoc.MustResult(aoc.Runner(run)) } + +type result struct { + valuePT1 int + valuePT2 int +} + +func (r result) String() string { return fmt.Sprintf("%#v", r) } + +func run(scan *bufio.Scanner) (*result, error) { + var m Map + + for scan.Scan() { + text := scan.Text() + m = append(m, []rune(text)) + } + + rows := len(m) + cols := len(m[0]) + + options := make([]int, 2*(rows+cols)+2) + i := 0 + for j:=0; j<=rows-1; j++ { + options[i+0] = runCycle(m, ray{[2]int{j, -1}, RT}) + options[i+1] = runCycle(m, ray{[2]int{j, cols}, LF}) + i+=2 + } + for j:=0; j<=cols-1; j++ { + options[i+0] = runCycle(m, ray{[2]int{-1, j}, DN}) + options[i+1] = runCycle(m, ray{[2]int{rows, j}, UP}) + i+=2 + } + + // fmt.Println(options) + return &result{valuePT1: options[0], valuePT2: aoc.Max(options[0], options[1:]...)}, nil +} + +type stack[T any] []T + +func (s *stack[T]) Push(v T) { + if s == nil { + panic("nil stack") + } + *s = append(*s, v) +} +func (s *stack[T]) Pop() T { + if s == nil || len(*s) == 0 { + panic("empty stack") + } + defer func() { *s = (*s)[:len(*s)-1] }() + return (*s)[len(*s)-1] +} + +var ( + ZERO = [2]int{0, -1} + + UP = [2]int{-1, 0} + DN = [2]int{1, 0} + LF = [2]int{0, -1} + RT = [2]int{0, 1} +) + +type ray struct { + pos [2]int + dir [2]int +} + +func (r *ray) next() [2]int { + r.pos[0] += r.dir[0] + r.pos[1] += r.dir[1] + return r.pos +} + +type Map [][]rune + +func (m *Map) Get(p [2]int) rune { + if p[0] < 0 || p[0] >= len((*m)) { + return 0 + } + if p[1] < 0 || p[1] >= len((*m)[0]) { + return 0 + } + + return (*m)[p[0]][p[1]] +} + + +func runCycle(m Map, r ray) int { + current := r + + s := stack[ray]{} + s.Push(current) + + energized := make(map[[2]int]bool) + // energized[current.pos] = true + + cycle := make(map[[4]int]bool) + + for len(s) > 0 { + current = s.Pop() + + r := m.Get(current.next()) + // fmt.Println("pos", current.pos, current.dir, string(r), len(s)) + if r == 0 { + continue + } + energized[current.pos] = true + v := [4]int{ + current.pos[0], + current.pos[1], + current.dir[0], + current.dir[1], + } + + if _, ok := cycle[v]; ok { + // fmt.Println("cycle") + continue + } + cycle[v] = true + + switch r { + case '|': + switch current.dir { + case UP, DN: + // pass + case LF, RT: + current.dir = UP + s.Push(ray{current.pos, DN}) + } + case '-': + switch current.dir { + case LF, RT: + // pass + case UP, DN: + current.dir = LF + s.Push(ray{current.pos, RT}) + } + case '/': + switch current.dir { + case UP: + current.dir = RT + case DN: + current.dir = LF + case LF: + current.dir = DN + case RT: + current.dir = UP + } + case '\\': + switch current.dir { + case UP: + current.dir = LF + case DN: + current.dir = RT + case LF: + current.dir = UP + case RT: + current.dir = DN + } + } + + s.Push(current) + } + + // for i := range m { + // for j := range m[i] { + // if v := energized[[2]int{i,j}]; v { + // fmt.Print("#") + // } else { + // fmt.Print(".") + // } + // } + // fmt.Println("") + // } + + return len(energized) +} \ No newline at end of file diff --git a/day16/main_test.go b/day16/main_test.go new file mode 100644 index 0000000..e10e1d1 --- /dev/null +++ b/day16/main_test.go @@ -0,0 +1,50 @@ +package main + +import ( + "bufio" + "bytes" + "testing" + + _ "embed" + + "github.com/matryer/is" +) + +//go:embed example.txt +var example []byte + +//go:embed input.txt +var input []byte + +func TestExample(t *testing.T) { + is := is.New(t) + scan := bufio.NewScanner(bytes.NewReader(example)) + + result, err := run(scan) + is.NoErr(err) + + t.Log(result) + is.Equal(result.valuePT1, 46) + is.Equal(result.valuePT2, 51) +} + +func TestSolution(t *testing.T) { + is := is.New(t) + scan := bufio.NewScanner(bytes.NewReader(input)) + + result, err := run(scan) + is.NoErr(err) + + t.Log(result) + + is.Equal(result.valuePT1, 8098) + is.Equal(result.valuePT2, 8335) +} + +func TestStack(t *testing.T) { + is := is.New(t) + + s := stack[int]{} + s.Push(5) + is.Equal(s.Pop(), 5) +} \ No newline at end of file diff --git a/tools_test.go b/tools_test.go new file mode 100644 index 0000000..e12e7a9 --- /dev/null +++ b/tools_test.go @@ -0,0 +1,80 @@ +package aoc_test + +import ( + "testing" + + "github.com/matryer/is" + aoc "go.sour.is/advent-of-code" +) + +func TestReverse(t *testing.T) { + is := is.New(t) + + is.Equal(aoc.Reverse([]int{1, 2, 3, 4}), []int{4, 3, 2, 1}) +} + +func TestLCM(t *testing.T) { + is := is.New(t) + + is.Equal(aoc.LCM([]int{}...), 0) + is.Equal(aoc.LCM(5), 5) + is.Equal(aoc.LCM(5, 3), 15) + is.Equal(aoc.LCM(5, 3, 2), 30) +} + +func TestReadStringToInts(t *testing.T) { + is := is.New(t) + + is.Equal(aoc.ReadStringToInts([]string{"1", "2", "3"}), []int{1, 2, 3}) +} + +func TestRepeat(t *testing.T) { + is := is.New(t) + + is.Equal(aoc.Repeat(5, 3), []int{5, 5, 5}) +} + +func TestPower2(t *testing.T) { + is := is.New(t) + + is.Equal(aoc.Power2(0), 1) + is.Equal(aoc.Power2(1), 2) + is.Equal(aoc.Power2(2), 4) +} + +func TestABS(t *testing.T) { + is := is.New(t) + + is.Equal(aoc.ABS(1), 1) + is.Equal(aoc.ABS(0), 0) + is.Equal(aoc.ABS(-1), 1) +} + +func TestTranspose(t *testing.T) { + is := is.New(t) + + is.Equal( + aoc.Transpose( + [][]int{ + {1, 1}, + {0, 0}, + {1, 1}, + }, + ), + [][]int{ + {1, 0, 1}, + {1, 0, 1}, + }, + ) +} + +func TestList(t *testing.T) { + is := is.New(t) + + lis := aoc.NewList[int](nil) + lis.Add(5, 0) + + a, _ := lis.Head().Value() + + is.Equal(a, 5) +}