Day 15 (refactoring)

day15/part2.py

@@ -1,22 +1,10 @@
 #!/usr/bin/env python3
 
-from collections import defaultdict
+from day15.common import get_target
 
 lines = (x.strip() for x in open("input.txt"))
-TARGET = 30000000
 
 for line in lines:
     print(line)
-    nums = defaultdict(list)
-    turn = 0
-    last_spoken = 0
-    for starting in map(int, line.split(",")):
-        nums[starting] = [turn]
-        turn += 1
-        last_spoken = starting
-    for t in range(turn, TARGET):
-        last = nums[last_spoken]
-        n = last[1] - last[0] if len(last) == 2 else 0
-        nums[n] = nums[n][-1:] + [t]
-        last_spoken = n
+    last_spoken = get_target(line, 30000000)
     print("->", last_spoken)