day 16

aoc2023/day16/README.md

@@ -0,0 +1,135 @@
+--- Day 16: The Floor Will Be Lava ---
+--------------------------------------
+
+With the beam of light completely focused *somewhere*, the reindeer leads you deeper still into the Lava Production Facility. At some point, you realize that the steel facility walls have been replaced with cave, and the doorways are just cave, and the floor is cave, and you're pretty sure this is actually just a giant cave.
+
+
+Finally, as you approach what must be the heart of the mountain, you see a bright light in a cavern up ahead. There, you discover that the beam of light you so carefully focused is emerging from the cavern wall closest to the facility and pouring all of its energy into a contraption on the opposite side.
+
+
+Upon closer inspection, the contraption appears to be a flat, two-dimensional square grid containing *empty space* (`.`), *mirrors* (`/` and `\`), and *splitters* (`|` and `-`).
+
+
+The contraption is aligned so that most of the beam bounces around the grid, but each tile on the grid converts some of the beam's light into *heat* to melt the rock in the cavern.
+
+
+You note the layout of the contraption (your puzzle input). For example:
+
+
+
+```
+.|...\....
+|.-.\.....
+.....|-...
+........|.
+..........
+.........\
+..../.\\..
+.-.-/..|..
+.|....-|.\
+..//.|....
+
+```
+
+The beam enters in the top-left corner from the left and heading to the *right*. Then, its behavior depends on what it encounters as it moves:
+
+
+* If the beam encounters *empty space* (`.`), it continues in the same direction.
+* If the beam encounters a *mirror* (`/` or `\`), the beam is *reflected* 90 degrees depending on the angle of the mirror. For instance, a rightward-moving beam that encounters a `/` mirror would continue *upward* in the mirror's column, while a rightward-moving beam that encounters a `\` mirror would continue *downward* from the mirror's column.
+* If the beam encounters the *pointy end of a splitter* (`|` or `-`), the beam passes through the splitter as if the splitter were *empty space*. For instance, a rightward-moving beam that encounters a `-` splitter would continue in the same direction.
+* If the beam encounters the *flat side of a splitter* (`|` or `-`), the beam is *split into two beams* going in each of the two directions the splitter's pointy ends are pointing. For instance, a rightward-moving beam that encounters a `|` splitter would split into two beams: one that continues *upward* from the splitter's column and one that continues *downward* from the splitter's column.
+
+
+Beams do not interact with other beams; a tile can have many beams passing through it at the same time. A tile is *energized* if that tile has at least one beam pass through it, reflect in it, or split in it.
+
+
+In the above example, here is how the beam of light bounces around the contraption:
+
+
+
+```
+>|<<<\....
+|v-.\^....
+.v...|->>>
+.v...v^.|.
+.v...v^...
+.v...v^..\
+.v../2\\..
+<->-/vv|..
+.|<<<2-|.\
+.v//.|.v..
+
+```
+
+Beams are only shown on empty tiles; arrows indicate the direction of the beams. If a tile contains beams moving in multiple directions, the number of distinct directions is shown instead. Here is the same diagram but instead only showing whether a tile is *energized* (`#`) or not (`.`):
+
+
+
+```
+######....
+.#...#....
+.#...#####
+.#...##...
+.#...##...
+.#...##...
+.#..####..
+########..
+.#######..
+.#...#.#..
+
+```
+
+Ultimately, in this example, `*46*` tiles become *energized*.
+
+
+The light isn't energizing enough tiles to produce lava; to debug the contraption, you need to start by analyzing the current situation. With the beam starting in the top-left heading right, *how many tiles end up being energized?*
+
+
+--- Part Two ---
+----------------
+
+As you try to work out what might be wrong, the reindeer tugs on your shirt and leads you to a nearby control panel. There, a collection of buttons lets you align the contraption so that the beam enters from *any edge tile* and heading away from that edge. (You can choose either of two directions for the beam if it starts on a corner; for instance, if the beam starts in the bottom-right corner, it can start heading either left or upward.)
+
+
+So, the beam could start on any tile in the top row (heading downward), any tile in the bottom row (heading upward), any tile in the leftmost column (heading right), or any tile in the rightmost column (heading left). To produce lava, you need to find the configuration that *energizes as many tiles as possible*.
+
+
+In the above example, this can be achieved by starting the beam in the fourth tile from the left in the top row:
+
+
+
+```
+.|<2<\....
+|v-v\^....
+.v.v.|->>>
+.v.v.v^.|.
+.v.v.v^...
+.v.v.v^..\
+.v.v/2\\..
+<-2-/vv|..
+.|<<<2-|.\
+.v//.|.v..
+
+```
+
+Using this configuration, `*51*` tiles are energized:
+
+
+
+```
+.#####....
+.#.#.#....
+.#.#.#####
+.#.#.##...
+.#.#.##...
+.#.#.##...
+.#.#####..
+########..
+.#######..
+.#...#.#..
+
+```
+
+Find the initial beam configuration that energizes the largest number of tiles; *how many tiles are energized in that configuration?*
+
+

aoc2023/day16/day16.py

@@ -0,0 +1,122 @@
+import os
+import sys
+from collections import Counter
+from itertools import chain
+
+sys.setrecursionlimit(5000)
+
+with open(os.path.join(os.path.dirname(__file__), "example.txt")) as example:
+    example_data = example.read().splitlines()
+    example_data = [list(n) for n in example_data]
+
+with open(os.path.join(os.path.dirname(__file__), "input.txt")) as example:
+    input_data = example.read().splitlines()
+    input_data = [list(n) for n in input_data]
+
+
+def init(data):
+    charged = [["."] * len(data[0]) for _ in range(len(data))]
+    direction = [["."] * len(data[0]) for _ in range(len(data))]
+    return data, charged, direction
+
+
+data, charged, direction = init(example_data)
+
+
+def bounce(i=0, j=0, succ=1, h=True, v=False):
+    if i >= len(data[0]) or i < 0 or j >= len(data) or j < 0:
+        return
+    if h:
+        if succ > 0 and direction[i][j] == ">":
+            return
+        elif succ > 0:
+            direction[i][j] = ">"
+        if succ < 0 and direction[i][j] == "<":
+            return
+        elif succ < 0:
+            direction[i][j] = "<"
+    else:
+        if succ > 0 and direction[i][j] == "v":
+            return
+        elif succ > 0:
+            direction[i][j] = "v"
+        if succ < 0 and direction[i][j] == "^":
+            return
+        elif succ < 0:
+            direction[i][j] = "^"
+
+    symbol = data[i][j]
+    charged[i][j] = "#"
+
+    if h:
+        if symbol == "." or symbol == "-":
+            bounce(i=i, j=j + succ, succ=succ, h=True, v=False)
+        elif symbol == "|":
+            bounce(i=i + succ, j=j, succ=succ, h=False, v=True)
+            bounce(i=i - succ, j=j, succ=-succ, h=False, v=True)
+        elif symbol == "\\":
+            bounce(i=i + succ, j=j, succ=succ, h=False, v=True)
+        elif symbol == "/":
+            bounce(i=i - succ, j=j, succ=-succ, h=False, v=True)
+        else:
+            raise Exception("PIDOR GNOY " + symbol)
+    elif v:
+        if symbol == "." or symbol == "|":
+            bounce(i=i + succ, j=j, succ=succ, h=False, v=True)
+        elif symbol == "\\":
+            bounce(i=i, j=j + succ, succ=succ, h=True, v=False)
+        elif symbol == "/":
+            bounce(i=i, j=j - succ, succ=-succ, h=True, v=False)
+        elif symbol == "-":
+            bounce(i=i, j=j + succ, succ=succ, h=True, v=False)
+            bounce(i=i, j=j - succ, succ=-succ, h=True, v=False)
+        else:
+            raise Exception("PIDOR GNOY " + symbol)
+    else:
+        raise Exception(f"PIDOR GNOY + {h} + {v}")
+
+
+# part 1 1
+data, charged, direction = init(example_data)
+bounce()
+print(c := Counter(chain(*charged))["#"])
+# part 1 2
+data, charged, direction = init(input_data)
+bounce()
+print(c := Counter(chain(*charged))["#"])
+
+# part 2 1
+m = 0
+for i in range(len(data)):
+    data, charged, direction = init(example_data)
+    bounce(i=i, j=0, succ=1, h=True, v=False)
+    m = max(m, Counter(chain(*charged))["#"])
+    data, charged, direction = init(example_data)
+    bounce(i=i, j=len(data[0]) - 1, succ=-1, h=True, v=False)
+    m = max(m, Counter(chain(*charged))["#"])
+for j in range(len(data[0])):
+    data, charged, direction = init(example_data)
+    bounce(i=0, j=j, succ=1, h=False, v=True)
+    m = max(m, Counter(chain(*charged))["#"])
+    data, charged, direction = init(example_data)
+    bounce(i=len(data) - 1, j=j, succ=-1, h=False, v=True)
+    m = max(m, Counter(chain(*charged))["#"])
+print(m)
+
+# part 2 2
+m = 0
+for i in range(len(data)):
+    data, charged, direction = init(input_data)
+    bounce(i=i, j=0, succ=1, h=True, v=False)
+    m = max(m, Counter(chain(*charged))["#"])
+    data, charged, direction = init(input_data)
+    bounce(i=i, j=len(data[0]) - 1, succ=-1, h=True, v=False)
+    m = max(m, Counter(chain(*charged))["#"])
+for j in range(len(data[0])):
+    data, charged, direction = init(input_data)
+    bounce(i=0, j=j, succ=1, h=False, v=True)
+    m = max(m, Counter(chain(*charged))["#"])
+    data, charged, direction = init(input_data)
+    bounce(i=len(data) - 1, j=j, succ=-1, h=False, v=True)
+    m = max(m, Counter(chain(*charged))["#"])
+print(m)

aoc2023/day16/example.txt

@@ -0,0 +1,10 @@
+.|...\....
+|.-.\.....
+.....|-...
+........|.
+..........
+.........\
+..../.\\..
+.-.-/..|..
+.|....-|.\
+..//.|....

aoc2023/day16/input.txt

@@ -0,0 +1,110 @@
+\-..../................\...../....|....|...........\...........\....../././............../...........|........
+....\.................\........./..........................................-..|.........../..../.........../..
+.|./........................./..........-.....|..../......../.........|......../.....|.................-....-.
+.--...............-.|........................\.......\...........|..........\.................................
+............|......|.........|.......\..\..../............./....-.../.........././............\.......\.......
+..|\.....|................/...........\.\....\\........\.........\..........|.......|...\.-....-.......|./....
+.........................../................................../../..|....../\........\................/...../.
+.../.......|./-..\....................................../....../...........-.....|.....|....../...............
+...-...\......-.........../..../.......\\..............-\....../....................|.....|...........\.......
+.|............-......../...........|........|...........-/.-..........\..............-..|............|-.......
+.................\/.........\/..../........./......../.-\........\..-.......|..............|..\..............|
+......./|../..........\/.............-........|........................\.................................-....
+....../......................................./.................--.....-.|......................\..../........
+.\.................|............|.//....|/.............|...........\|..........-..........-................-..
+.../................|........|...-.................\....|....|....|..............................|............
+|.........-\...-.-......./.............-.-/...............-.\..../........................./.........\....\...
+......|........./............\.../\.....-...|...............\.....|................................\..........
+.........................................\......\.........................-......\...-......../......./\-./...
+....../.../......-..................................................../|.......|........................\.....
+....................../\/..........................|.......................................|..................
+|.......\.......................|..-..\......\..\...............................................|.......|.....
+................../........-...|..........................|............-........\.......|...........\.........
+.............................|.............../.......\....|......................./............./......./..../
+.\.............-..-...|...-.-....|......./.........../...../........./....................\././.....|.........
+..-..../....\......................|............................................./.\.../..........|...........
+....-.........-.......\......-.............../.............|....|......./.......\..-.\.....\..-...............
+..................../-.......|..../....-/...|...........-//..............-./...........-...........././.......
+..............................................|.-........|...........\..|/|..-.....-...../......../...........
+...............................\.../-..|......|......./.\................--............-........../.....\./...
+-.............|.-......................................................|.....\.|....-.....|......\|........\..
+-....\.....-...-.............../...........-.................................../.|.\........-.................
+....-....\.-./.../..../........\..-.............|...................|.........\.....\................../..\|..
+...................||..|.......|.....\.../.....-../.....-.|............/.......-.........|........./..-......|
+.........\........-....-..-.....|..............|.........-.\|.\|.........\....|.........\.\\...........|.\.|..
+..........................|..|.../../...-...............-....../....................|...............-.........
+...../|........|./..................-....\.|...........|..............................\./.../\................
+............................../..\..\............-.../.........-...../.......-..\...|....../..-|-.............
+..........-...../.......-................-...........|..\|..|.\.............../.-.../.........................
+............................../..-.....-...../............................\........|...................\......
+/./.......-|...-...\/.......-................/.....-....\...../..............\../............-................
+.............\...........................-...|................................\./..-............-........|/|./
+......................./..-.......\........./...\..|..../.....................-.....-\|...................../.
+.....|...-....-..-/....|.|................................../............/............./..\........./.....-...
+.....|................\.../.....\..........................\|...........-.\.|./..-.....\....../....\....|.....
+......................\................-............../\....||....-...-.....|......|-.......................-.
+.................................|..........-......|.....-......-........|.|...../.........-......\..../|.....
+................../.......\.....................................|......................................-......
+......\.....\..../..-................-......................-..................................-...--.-.-.....
+...........-...........\.............................................................-.-........-.............
+.|..............\./........................../................./..-....-........|-/.../.......-........\......
+.-...-/...|.|.../..-........../..............................|.../..........\.............-.............../.|.
+.............\..|.....\.....\.................-....\...................|../.-......|.......\..\.../....../....
+/........|....\.........\.....|..........\./..\.........-.-.......-.....|....../.......\-\..-..../............
+..........|...|\...|............\......\.............-.|...........-.....................|..................\-
+............\.\...../../..........................-.|...................../........\..-....-............/.....
+............./.../|...\.......-...................\....|....-\..........|/.-//.\.......\..-.\.....\...\../....
+.-..............|.....|.........|......|\................//...|...........................\...\...-...........
+......-...\../...........................|...............\.||.........................-..............-........
+.......-............|...|....-...\....|...|........|..../........................................|.........\..
+........\.|.\....\.\.|...........|......./.....................\.../.............-...|./...........\..........
+........./......-.........-................|.../...............-.............../|\..........-./...............
+...\..........\......./........\........./.../........../........-......-........-......................./....
+............./../........../.........../................-...............-./....-............................-.
+..\/.|....................|.....\................-...............--...................../............./.......
+.............................-....-........................................................|..........|..|....
+./................................/............../..................................................|..\.../..
+...............-...............|...............................-.../.-.|..-........../.../....................
+.......|...\.............\...\./................-......\..\././........-.............../.................../..
+..../..\...-..............|./........../....../...../........\....../......-....../..|......-.................
+.............................--./..........-....\...........|.............................../.................
+.........../...............-....-|......./..........-..|.....................-......\.\.................../...
+.-.|...-.\................/../......\...........|.......-...........-.........................................
+./........\-........../..........././.............|...\................-............|....................|....
+.........|..\||............|.-..\.................-.....-.......-..|...........\................|......../....
+....-.................\...-......\.........|...........\...../.....\.......-....................|/......-.....
+......-...|...........\.....-....................-.........../.....\...\......................................
+../.....\....|..-..\.............../..........................|..\-.\........................|....\....-.\..|.
+/--....|.../.........\./.................|..-.........|....-..............................-.-../..............
+.\...............//......-........./.....................................\|...\......|.............../........
+..........................\.........\..................\..................\.................-...|./...........
+.......--................../........................|........................\..........|.....................
+.........-............................-./......./.|..................|.........\...............|..............
+...../..\................../...-.....|...........\|.......|...........\........./..|....-...........-.........
+....-.....\..................|......\.../...\........./.....\..............................\./................
+.\.......\......|......|...................\.....-../...........................-.....-...........-/..........
+.....\........./......./..\....................../../.......-....../.....|.........-........|.....|\/.........
+........................\........./....................../../............-.........\........|\......\-......\.
+..........................-./.....\../..\........./..............\.........\.......|...../......|.........../.
+........................|......-\...-..........|.......................-../.....|..-......................|...
+.....//.../......................\...\............................................./..\.|....../..........|...
+-...\.|.../....................................\..................-.-...|..|................-..|......|.....|.
+.....\..-.......................|...|..............\.......\..........\....-../......................|.-......
+/\.....-..\.........-...|.....|.......\../../................../.-.....\......-.|............/.........-......
+....................................././............\........|....|........................../......../....|..
+..............|................|....\........./................||...........|................/-....\...-......
+............/|.....|.........\..............|.....|........./....|.............../......\............./..\....
+-...../.....\.........-...|............./................./........../......./............./.....|........\...
+...|./......................................\.......|...../............-...............\.././...\.............
+..\...............\....../..\....|............................\.....|....\....-...................-...........
+..............\.....\......./............--..................||.......\..-../.|.............|....../.-....\.-.
+/..............-.......................-..........................\............\....................|........|
+......-...................../..............................\...|.....\../..............|...|..................
+.....-.....\...../.-........................-.........|..........\............\../.......|........./.-.|.....\
+....-................|...././..\...-......./..../.........-........|/......../..........\......|/..\......-...
+./...|..\....|./......\....../..........-.|..........\.............-..........\|.\............................
+.....|..|..../........................../......|./../...-................-|......\............................
+./|......|................|..-.............../\....-............./-...................-.|.........|..........|
+............................................|.\........-.\..../..|..................-.........-....-.-./......
+...../............................\...................-.-...-../...........|..|....\...................-.....\
+......./...\../..........-.................-.........................-........../...-......\..................