feat: add solution 2021 11th

Author: Garcel

2021/11/input

@@ -0,0 +1,10 @@
+4341347643
+5477728451
+2322733878
+5453762556
+2718123421
+4237886115
+5631617114
+2217667227
+4236581255
+4482627641

2021/11/puzzle1.py

@@ -0,0 +1,75 @@
+import os
+import sys
+import typing
+
+
+def main():
+ with open(os.path.join(sys.path[0], 'input'), 'r') as input_file:
+ print(puzzle1(input_file))
+
+
+def puzzle1(input_file: typing.TextIO, steps: int = 100):
+ rows = []
+
+ for row in input_file:
+ rows.append([int(number) for number in row.strip()])
+
+ grid = OctopusGrid(rows)
+
+ for i in range(0, steps):
+ grid.apply_step()
+
+ return grid.get_flashes_count()
+
+
+class OctopusGrid:
+ grid: [[int]]
+ flashes: int
+
+ def __init__(self, grid: [[int]]):
+ self.grid = grid
+ self.flashes = 0
+
+ def apply_step(self):
+ seen = set()
+
+ for row_number in range(0, len(self.grid)):
+ for column_number in range(0, len(self.grid[row_number])):
+ self._increase_octopus_energy(row_number, column_number, seen)
+
+ def _increase_octopus_energy(self, row_number: int, column_number: int, seen: set):
+ if row_number < 0 or row_number >= len(self.grid) or \
+ column_number < 0 or column_number >= len(self.grid[row_number]):
+ return
+
+ position = (row_number, column_number)
+ if position in seen:
+ return
+
+ self.grid[row_number][column_number] += 1
+
+ if not self.grid[row_number][column_number] > 9:
+ return
+
+ seen.add(position)
+ self._flash(row_number, column_number, seen)
+
+ def _flash(self, row_number: int, column_number: int, seen: set):
+ self.flashes += 1
+ self.grid[row_number][column_number] = 0
+
+ self._increase_octopus_energy(row_number, column_number - 1, seen)
+ self._increase_octopus_energy(row_number, column_number + 1, seen)
+ self._increase_octopus_energy(row_number - 1, column_number, seen)
+ self._increase_octopus_energy(row_number + 1, column_number, seen)
+ self._increase_octopus_energy(row_number - 1, column_number - 1, seen)
+ self._increase_octopus_energy(row_number - 1, column_number + 1, seen)
+ self._increase_octopus_energy(row_number + 1, column_number - 1, seen)
+ self._increase_octopus_energy(row_number + 1, column_number + 1, seen)
+
+ def get_flashes_count(self):
+ return self.flashes
+
+
+if __name__ == "__main__":
+ main()

2021/11/puzzle2.py

@@ -0,0 +1,84 @@
+import os
+import sys
+import typing
+
+
+def main():
+ with open(os.path.join(sys.path[0], 'input'), 'r') as input_file:
+ print(puzzle2(input_file))
+
+
+def puzzle2(input_file: typing.TextIO):
+ rows = []
+
+ for row in input_file:
+ rows.append([int(number) for number in row.strip()])
+
+ grid = OctopusGrid(rows)
+
+ while not grid.sync():
+ grid.apply_step()
+
+ return grid.get_steps_count()
+
+
+class OctopusGrid:
+ grid: [[int]]
+ flashes: int
+ steps: int
+
+ def __init__(self, grid: [[int]]):
+ self.grid = grid
+ self.flashes = 0
+ self.steps = 0
+
+ def apply_step(self):
+ self.steps += 1
+ seen = set()
+
+ for row_number in range(0, len(self.grid)):
+ for column_number in range(0, len(self.grid[row_number])):
+ self._increase_octopus_energy(row_number, column_number, seen)
+
+ def _increase_octopus_energy(self, row_number: int, column_number: int, seen: set):
+ if row_number < 0 or row_number >= len(self.grid) or \
+ column_number < 0 or column_number >= len(self.grid[row_number]):
+ return
+
+ position = (row_number, column_number)
+ if position in seen:
+ return
+
+ self.grid[row_number][column_number] += 1
+
+ if not self.grid[row_number][column_number] > 9:
+ return
+
+ seen.add(position)
+ self._flash(row_number, column_number, seen)
+
+ def _flash(self, row_number: int, column_number: int, seen: set):
+ self.flashes += 1
+ self.grid[row_number][column_number] = 0
+
+ self._increase_octopus_energy(row_number, column_number - 1, seen)
+ self._increase_octopus_energy(row_number, column_number + 1, seen)
+ self._increase_octopus_energy(row_number - 1, column_number, seen)
+ self._increase_octopus_energy(row_number + 1, column_number, seen)
+ self._increase_octopus_energy(row_number - 1, column_number - 1, seen)
+ self._increase_octopus_energy(row_number - 1, column_number + 1, seen)
+ self._increase_octopus_energy(row_number + 1, column_number - 1, seen)
+ self._increase_octopus_energy(row_number + 1, column_number + 1, seen)
+
+ def get_flashes_count(self) -> int:
+ return self.flashes
+
+ def get_steps_count(self) -> int:
+ return self.steps
+
+ def sync(self) -> bool:
+ return sum([sum(row) for row in self.grid]) == 0
+
+
+if __name__ == "__main__":
+ main()

2021/11/test_puzzle1.py

@@ -0,0 +1,37 @@
+import io
+import os
+import sys
+
+import pytest
+from puzzle1 import puzzle1
+
+
+@pytest.mark.parametrize(
+ "test_input,expected", [
+ (
+ """5483143223
+ 2745854711
+ 5264556173
+ 6141336146
+ 6357385478
+ 4167524645
+ 2176841721
+ 6882881134
+ 4846848554
+ 5283751526""",
+ 1656
+ )
+ ])
+def test_puzzle1_sample(test_input, expected):
+ assert puzzle1(io.StringIO(test_input)) == expected
+
+
+@pytest.mark.parametrize(
+ "test_input,expected", [
+ (
+ open(os.path.join(sys.path[0], 'input'), 'r'),
+ 1697
+ )
+ ])
+def test_puzzle1(test_input, expected):
+ assert puzzle1(test_input) == expected

2021/11/test_puzzle2.py

@@ -0,0 +1,37 @@
+import io
+import os
+import sys
+
+import pytest
+from puzzle2 import puzzle2
+
+
+@pytest.mark.parametrize(
+ "test_input,expected", [
+ (
+ """5483143223
+ 2745854711
+ 5264556173
+ 6141336146
+ 6357385478
+ 4167524645
+ 2176841721
+ 6882881134
+ 4846848554
+ 5283751526""",
+ 195
+ )
+ ])
+def test_puzzle2_sample(test_input, expected):
+ assert puzzle2(io.StringIO(test_input)) == expected
+
+
+@pytest.mark.parametrize(
+ "test_input,expected", [
+ (
+ open(os.path.join(sys.path[0], 'input'), 'r'),
+ 344
+ )
+ ])
+def test_puzzle2(test_input, expected):
+ assert puzzle2(test_input) == expected