Day 15 (static array length 14s -> 10s)

day15/common.py

@@ -2,7 +2,7 @@ from functools import reduce
 
 
 def yield_numbers(line: str, target: int):
-    nums = {}
+    nums = [None]*target
     turn = 0
     last_spoken = 0
     for starting in map(int, line.split(",")):
@@ -11,7 +11,7 @@ def yield_numbers(line: str, target: int):
         last_spoken = starting
         yield turn, last_spoken, nums
     for t in range(turn, target):
-        n = t - nums[last_spoken] if last_spoken in nums else 0
+        n = t - nums[last_spoken] if nums[last_spoken] is not None else 0
         nums[last_spoken] = t
         last_spoken = n
         yield t+1, last_spoken, nums