[2024/11] p2 solved

2024/11/script.py

@@ -1,7 +1,7 @@
 from GhostyUtils import aoc
 import math
 from functools import cache
-from collections import Counter
+from collections import Counter, defaultdict
 
 
 @cache
@@ -18,29 +18,45 @@ def split(value: int) -> tuple[int, int]:
     return (l_value, r_value)
 
 
-def blink(stones: list[int]) -> list[int]:
-    new_stones = []
-    for i, stone in enumerate(stones):
+def blink(stones: dict[int, int]) -> dict[int, int]:
+    new_stones = defaultdict(int)
+    for stone, count in stones.items():
         if stone == 0:
-            new_stones.append(1)
+            new_stones[1] += count
         elif num_digits(stone) % 2 == 0:
-            new_stones.extend(split(stone))
+            l_stone, r_stone = split(stone)
+            new_stones[l_stone] += count
+            new_stones[r_stone] += count
         else:
-            new_stones.append(stone * 2024)
+            new_stones[stone * 2024] += count
 
     return new_stones
 
 
-def main():
-    stones = [int(stone) for stone in aoc.read().split()]
-    if aoc.args.verbose:
-        print(stones)
-
-    for _ in range(25):
+def blink_n_times(stones: dict[int, int], n: int) -> dict[int, int]:
+    for _ in range(n):
         stones = blink(stones)
+
         if aoc.args.verbose:
             print(stones)
-    print(f"p1: {len(stones)}")
+
+    return stones
+
+
+def num_stones(stones: dict[int, int]) -> int:
+    return sum(count for _, count in stones.items())
+
+
+def main():
+    start_stones = Counter([int(stone) for stone in aoc.read().split()])
+    if aoc.args.verbose:
+        print(start_stones)
+
+    stones = blink_n_times(start_stones, 25)
+    print(f"p1: {num_stones(stones)}")
+
+    stones = blink_n_times(start_stones, 75)
+    print(f"p2: {num_stones(stones)}")
 
 
 if __name__ == "__main__":

README.md

@@ -3,10 +3,10 @@ My solutions to the yearly Advents of Code
 
 <!-- AOC TILES BEGIN -->
 <h1 align="center">
-  Advent of Code - 182/472 ⭐
+  Advent of Code - 183/472 ⭐
 </h1>
 <h1 align="center">
-  2024 - 21 ⭐ - Python
+  2024 - 22 ⭐ - Python
 </h1>
 <a href="2024/1/script.py">
   <img src=".aoc_tiles/tiles/2024/01.png" width="161px">