From d8e68f10f37802aa5c81805d994a8133c621520a Mon Sep 17 00:00:00 2001 From: kamoshi <18511281+kamoshi@users.noreply.github.com> Date: Tue, 15 Dec 2020 08:42:21 +0100 Subject: [PATCH] Update day15.py --- 2020/Python/day15.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/2020/Python/day15.py b/2020/Python/day15.py index cea131f..09fe119 100644 --- a/2020/Python/day15.py +++ b/2020/Python/day15.py @@ -6,7 +6,7 @@ def parse_data() -> list[int]: return numbers -def solve_p1(data: list[int]) -> int: +def solve(data: list[int], which_number: int) -> int: turns = {} curr_turn = 1 last_num = 0 @@ -23,7 +23,8 @@ def solve_p1(data: list[int]) -> int: last_num = number curr_turn += 1 - for i in range(curr_turn, 2021): + _limit = which_number + 1 + for i in range(curr_turn, _limit): if last_num in turns: (before, last) = turns[last_num] if before != -1: @@ -37,8 +38,12 @@ def solve_p1(data: list[int]) -> int: return last_num +def solve_p1(data: list[int]) -> int: + return solve(data, which_number=2020) + + def solve_p2(data: list[int]) -> int: - pass + return solve(data, which_number=30000000) DATA = parse_data()