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()