advent-of-code/2020/Python/day01.py

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