22 lines
569 B
Python
22 lines
569 B
Python
|
#! /usr/bin/env python3
|
||
|
|
from itertools import product
|
||
|
|
|
||
|
|
|
||
|
|
def part1(inp):
|
||
|
|
inp = [int(x) for x in inp]
|
||
|
|
result_pairs = [x for x in list(product(inp, inp)) if sum(x) == 2020]
|
||
|
|
print(result_pairs)
|
||
|
|
print(result_pairs[0][0] * result_pairs[0][1])
|
||
|
|
|
||
|
|
|
||
|
|
def part2(inp):
|
||
|
|
inp = [int(x) for x in inp]
|
||
|
|
result_pairs = [x for x in list(product(inp, repeat=3)) if sum(x) == 2020]
|
||
|
|
print(result_pairs)
|
||
|
|
print(result_pairs[0][0] * result_pairs[0][1] * result_pairs[0][2])
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
with open("input.txt") as f:
|
||
|
|
part2(f.readlines())
|