2021 day 11.

src/holt59/aoc/2021/day11.py

@@ -1,7 +1,66 @@
+import itertools as it
 from typing import Any, Iterator
 
 from ..base import BaseSolver
 
 
+def do_step(values: list[list[int]]) -> tuple[list[list[int]], set[tuple[int, int]]]:
+    values = [[c + 1 for c in r] for r in values]
+    flashed: set[tuple[int, int]] = set()
+
+    while True:
+        found = False
+
+        for i_row, row in enumerate(values):
+            for i_col, col in enumerate(row):
+                if col <= 9 or (i_row, i_col) in flashed:
+                    continue
+
+                found = True
+                flashed.add((i_row, i_col))
+                for dr, dc in it.product((-1, 0, 1), repeat=2):
+                    if 0 <= i_row + dr < len(values) and 0 <= i_col + dc < len(
+                        values[0]
+                    ):
+                        values[i_row + dr][i_col + dc] += 1
+
+        if not found:
+            break
+
+    for i, j in flashed:
+        values[i][j] = 0
+
+    return values, flashed
+
+
 class Solver(BaseSolver):
-    def solve(self, input: str) -> Iterator[Any]: ...
+    def print_grid(self, values: list[list[int]], flashed: set[tuple[int, int]]):
+        for i_row, row in enumerate(values):
+            s_row = ""
+            for i_col, col in enumerate(row):
+                if (i_row, i_col) in flashed:
+                    s_row += f"\033[0;31m{col}\033[0;00m"
+                else:
+                    s_row += str(col)
+            self.logger.info(s_row)
+        self.logger.info("")
+
+    def solve(self, input: str) -> Iterator[Any]:
+        values_0 = [[int(c) for c in r] for r in input.splitlines()]
+
+        values = values_0
+        total_flashed: int = 0
+        for _ in range(100):
+            values, flashed = do_step(values)
+            total_flashed += len(flashed)
+
+        yield total_flashed
+
+        n_cells = len(values) * len(values[0])
+        flashed: set[tuple[int, int]] = set()
+        values, step = values_0, 0
+        while len(flashed) != n_cells:
+            values, flashed = do_step(values)
+            step += 1
+
+        yield step

src/holt59/aoc/inputs/holt59/2021/day11.txt

@@ -0,0 +1,10 @@
+4738615556
+6744423741
+2812868827
+8844365624
+4546674266
+4518674278
+7457237431
+4524873247
+3153341314
+3721414667

src/holt59/aoc/inputs/tests/2021/day11.txt

@@ -0,0 +1,10 @@
+5483143223
+2745854711
+5264556173
+6141336146
+6357385478
+4167524645
+2176841721
+6882881134
+4846848554
+5283751526