kamov/advent-of-code
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Update day15.py

Browse source
This commit is contained in:
kamoshi 2020-12-15 08:42:21 +01:00 committed by GitHub
parent 59884f4ec9
commit d8e68f10f3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
2020/Python/day15.py
View file

@ -6,7 +6,7 @@ def parse_data() -> list[int]:
return numbers return numbers
def solve_p1(data: list[int]) -> int: def solve(data: list[int], which_number: int) -> int:
turns = {} turns = {}
curr_turn = 1 curr_turn = 1
last_num = 0 last_num = 0
@ -23,7 +23,8 @@ def solve_p1(data: list[int]) -> int:
last_num = number last_num = number
curr_turn += 1 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: if last_num in turns:
(before, last) = turns[last_num] (before, last) = turns[last_num]
if before != -1: if before != -1:
@ -37,8 +38,12 @@ def solve_p1(data: list[int]) -> int:
return last_num return last_num
def solve_p1(data: list[int]) -> int:
return solve(data, which_number=2020)
def solve_p2(data: list[int]) -> int: def solve_p2(data: list[int]) -> int:
pass return solve(data, which_number=30000000)
DATA = parse_data() DATA = parse_data()