25 lines
597 B
Python
25 lines
597 B
Python
def solve_p1():
|
|
numbers = []
|
|
with open("input.txt") as f:
|
|
for line in f:
|
|
num = int(line)
|
|
opp = 2020-num
|
|
if opp in numbers:
|
|
print(num*opp)
|
|
numbers.append(num)
|
|
|
|
|
|
def solve_p2():
|
|
numbers = []
|
|
with open("input.txt") as f:
|
|
for line in f:
|
|
numbers.append(int(line))
|
|
combos = [((a, b, c), a*b*c) if a+b+c == 2020 else None for (a, b, c) in itertools.combinations(numbers, 3)]
|
|
for combo in combos:
|
|
if combo is not None:
|
|
print(combo)
|
|
|
|
|
|
solve_p1():
|
|
solve_p2()
|