[2024/11] add --num_blinks script arg

2024/11/script.py

@@ -4,6 +4,10 @@
 from collections import Counter, defaultdict
 
 
+aoc.argparser.add_argument("-n", "--num_blinks", default=0, type=int,
+                           help="number of times to blink")
+
+
 @cache
 def num_digits(value: int) -> int:
     return int(math.log10(value)) + 1
@@ -48,6 +52,11 @@ def main():
     if aoc.args.verbose:
         print(start_stones)
 
+    if aoc.args.num_blinks > 0:
+        stones = blink_n_times(start_stones, aoc.args.num_blinks)
+        print(f"{aoc.args.num_blinks} blinks: {sum(stones.values())}")
+        exit(0)
+
     stones = blink_n_times(start_stones, 25)
     print(f"p1: {sum(stones.values())}")