19 lines
418 B
Python
19 lines
418 B
Python
|
with open('input01', 'r') as file:
|
||
|
|
data = sorted([int(i) for i in file.readlines()])
|
||
|
|
|
||
|
|
def find_pair_sum(total, data):
|
||
|
|
for x in data:
|
||
|
|
y = total - x
|
||
|
|
if y in data:
|
||
|
|
return x, y
|
||
|
|
return None, None
|
||
|
|
|
||
|
|
x, y = find_pair_sum(2020, data)
|
||
|
|
print(f'Part 1: {x}*{y} = {x*y}')
|
||
|
|
|
||
|
|
for idx, x in enumerate(data):
|
||
|
|
y, z = find_pair_sum(2020-x, data[idx+1:])
|
||
|
|
if y:
|
||
|
|
print(f'Part 2: {x}*{y}*{z} = {x*y*z}')
|
||
|
|
break
|