inputs/day06-example.txt

@@ -0,0 +1 @@
+3,4,3,1,2

inputs/day06.txt

@@ -0,0 +1 @@
+1,3,4,1,5,2,1,1,1,1,5,1,5,1,1,1,1,3,1,1,1,1,1,1,1,2,1,5,1,1,1,1,1,4,4,1,1,4,1,1,2,3,1,5,1,4,1,2,4,1,1,1,1,1,1,1,1,2,5,3,3,5,1,1,1,1,4,1,1,3,1,1,1,2,3,4,1,1,5,1,1,1,1,1,2,1,3,1,3,1,2,5,1,1,1,1,5,1,5,5,1,1,1,1,3,4,4,4,1,5,1,1,4,4,1,1,1,1,3,1,1,1,1,1,1,3,2,1,4,1,1,4,1,5,5,1,2,2,1,5,4,2,1,1,5,1,5,1,3,1,1,1,1,1,4,1,2,1,1,5,1,1,4,1,4,5,3,5,5,1,2,1,1,1,1,1,3,5,1,2,1,2,1,3,1,1,1,1,1,4,5,4,1,3,3,1,1,1,1,1,1,1,1,1,5,1,1,1,5,1,1,4,1,5,2,4,1,1,1,2,1,1,4,4,1,2,1,1,1,1,5,3,1,1,1,1,4,1,4,1,1,1,1,1,1,3,1,1,2,1,1,1,1,1,2,1,1,1,1,1,1,1,2,1,1,1,1,1,1,4,1,1,1,1,1,1,1,1,1,1,1,1,1,1,2,1,1,2,5,1,2,1,1,1,1,1,1,1,1,1

puzzles/day06.py

@@ -0,0 +1,77 @@
+"""
+Advent of Code 2021 - day 06
+
+Run with:
+    python puzzles/day06.py inputs/day06.txt
+"""
+
+import pathlib
+import sys
+
+from collections import Counter
+from typing import List, Tuple
+
+
+class Reproduce:
+    times = None
+
+    def __init__(self, times: List[int]) -> None:
+        self.times = times
+
+    def start(self, days: int = 80) -> None:
+        counts = Counter(self.times)
+        for _ in range(days):
+            next_counts = Counter()  # Can't modify while iterating
+
+            for count_key, count in counts.items():
+                if count_key == 0:
+                    next_counts[8] += count  # Add the new fish
+                    next_counts[6] += count  # Move the parents back to 6 days
+                else:
+                    next_counts[count_key - 1] += count  # Move count to next day count
+
+            counts = next_counts
+        self.time_counts = counts
+
+    def count_fish(self) -> int:
+        return sum(self.time_counts.values())
+
+
+def part1(inputs: List[int]) -> int:
+    school = Reproduce(inputs)
+    school.start(days=80)
+    return school.count_fish()
+
+
+def part2(inputs: List[int]) -> int:
+    school = Reproduce(inputs)
+    school.start(days=256)
+    return school.count_fish()
+
+
+def parse(inputs: str) -> List[int]:
+    """Parse the input string"""
+    return [int(value) for value in inputs.split(",")]
+
+
+def solve(path: str) -> Tuple[int, int]:
+    """Solve the puzzle"""
+    puzzle_input = parse(pathlib.Path(path).read_text().strip())
+    part1_result = part1(puzzle_input)
+    part2_result = part2(puzzle_input)
+
+    return part1_result, part2_result
+
+
+def main() -> None:
+    for path in sys.argv[1:]:
+        print(f"Input File: {path}")
+
+        part1_result, part2_result = solve(path)
+
+        print(f"Part 1 Result: {part1_result}")
+        print(f"Part 2 Result: {part2_result}")
+
+
+if __name__ == "__main__":
+    main()

puzzles/test_day06.py

@@ -0,0 +1,32 @@
+import pathlib
+import pytest
+import day06 as aoc
+
+INPUTS_DIR = f"{pathlib.Path(__file__).parent.parent}/inputs"
+
+@pytest.fixture
+def example_data():
+    input_path = f"{INPUTS_DIR}/day06-example.txt"
+    return aoc.parse(pathlib.Path(input_path).read_text().strip())
+
+
+@pytest.fixture
+def day06_data():
+    input_path = f"{INPUTS_DIR}/day06.txt"
+    return aoc.parse(pathlib.Path(input_path).read_text().strip())
+
+
+def test_example1(example_data):
+    assert aoc.part1(example_data) == 5934
+
+
+def test_example2(example_data):
+    assert aoc.part2(example_data) == 26984457539
+
+
+def test_part1(day06_data):
+    assert aoc.part1(day06_data) == 386755
+
+
+def test_part2(day06_data):
+    assert aoc.part2(day06_data) == 1732731810807