aoc

Advent of Code
git clone https://git.pngu.org/aoc
Log | Files | Refs | README

1_2020.py (339B)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
import sys
import fileinput

X = [int(line) for line in fileinput.input()]
n = len(X)
for i in range(n):
    for j in range(i + 1, n):
        if X[i] + X[j] == 2020:
            print("Part 1:", X[i] * X[j])
        for k in range(j + 1, n):
            if X[i] + X[j] + X[k] == 2020:
                print("Part 2:", X[i] * X[j] * X[k])