formatting and cosmetic updates to 2020.
This commit is contained in:
parent
84c4cf9991
commit
cd75e58f77
28 changed files with 320 additions and 287 deletions
|
|
@ -2,41 +2,43 @@
|
|||
|
||||
import pathlib
|
||||
import sys
|
||||
import re
|
||||
from dataclasses import dataclass
|
||||
|
||||
|
||||
def parse(puzzle_input):
|
||||
"""Parse input"""
|
||||
return [int(string) for string in puzzle_input.splitlines()]
|
||||
|
||||
def part1(data:list[int]):
|
||||
|
||||
def part1(data: list[int]):
|
||||
"""Solve part 1"""
|
||||
data.sort()
|
||||
while True:
|
||||
if len(data)<2:
|
||||
raise ValueError('no match found')
|
||||
s = data[0]+data[-1]
|
||||
if s>2020:
|
||||
if len(data) < 2:
|
||||
raise ValueError("no match found")
|
||||
s = data[0] + data[-1]
|
||||
if s > 2020:
|
||||
data.pop(-1)
|
||||
elif s<2020:
|
||||
elif s < 2020:
|
||||
data.pop(0)
|
||||
else:
|
||||
return data[0]*data[-1]
|
||||
return data[0] * data[-1]
|
||||
|
||||
|
||||
def part2(data):
|
||||
"""Solve part 2"""
|
||||
data.sort()
|
||||
while True:
|
||||
if len(data)<3:
|
||||
raise ValueError('no match found')
|
||||
if data[0]+data[1]+data[-1]>2020:
|
||||
if len(data) < 3:
|
||||
raise ValueError("no match found")
|
||||
if data[0] + data[1] + data[-1] > 2020:
|
||||
data.pop(-1)
|
||||
elif data[0]+data[-1]+data[-2]<2020:
|
||||
elif data[0] + data[-1] + data[-2] < 2020:
|
||||
data.pop(0)
|
||||
elif data[0]+data[1]+data[-1]==2020:
|
||||
return data[0]*data[1]*data[-1]
|
||||
elif data[0] + data[1] + data[-1] == 2020:
|
||||
return data[0] * data[1] * data[-1]
|
||||
else:
|
||||
return data[0]*data[-1]*data[-2]
|
||||
return data[0] * data[-1] * data[-2]
|
||||
|
||||
|
||||
def solve(puzzle_input):
|
||||
"""Solve the puzzle for the given input"""
|
||||
|
|
@ -46,6 +48,7 @@ def solve(puzzle_input):
|
|||
|
||||
return solution1, solution2
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
for path in sys.argv[1:]:
|
||||
print(f"{path}:")
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue