diff --git a/day15/part1.py b/day15/part1.py index 9d8e34e..ff94a54 100644 --- a/day15/part1.py +++ b/day15/part1.py @@ -3,11 +3,11 @@ from collections import defaultdict lines = (x.strip() for x in open("input.txt")) -nums = defaultdict(list) TARGET = 2020 for line in lines: print(line) + nums = defaultdict(list) turn = 0 last_spoken = 0 for starting in map(int, line.split(",")): @@ -16,10 +16,7 @@ for line in lines: last_spoken = starting for t in range(turn, TARGET): last = nums[last_spoken] - if len(last) == 2: - n = last[1] - last[0] - else: - n = 0 + n = last[1] - last[0] if len(last) == 2 else 0 nums[n] = nums[n][-1:] + [t] last_spoken = n print("->", last_spoken) diff --git a/day15/part2.py b/day15/part2.py index 63a1973..1acd8a7 100644 --- a/day15/part2.py +++ b/day15/part2.py @@ -3,11 +3,11 @@ from collections import defaultdict lines = (x.strip() for x in open("input.txt")) -nums = defaultdict(list) TARGET = 30000000 for line in lines: print(line) + nums = defaultdict(list) turn = 0 last_spoken = 0 for starting in map(int, line.split(",")): @@ -16,10 +16,7 @@ for line in lines: last_spoken = starting for t in range(turn, TARGET): last = nums[last_spoken] - if len(last) == 2: - n = last[1] - last[0] - else: - n = 0 + n = last[1] - last[0] if len(last) == 2 else 0 nums[n] = nums[n][-1:] + [t] last_spoken = n print("->", last_spoken)