advent2020/one.py

12 lines
321 B
Python
Executable File

#!/usr/bin/env python3
with open('input1') as f:
l = f.readlines()
n = [int(x) for x in l]
a, b = next(filter(lambda x: sum(x) == 2020, ((a, b) for a in n for b in n)))
print(a * b)
a, b, c = next(filter(lambda x: sum(x) == 2020, ((a, b, c) for a in n for b in n for c in n)))
print(a * b * c)