aoc2020/10/1.py

13 lines
249 B
Python

with open("input") as f:
content = [int(x.strip()) for x in f]
content.sort()
diffs = []
for i in range(len(content) -1 ):
diffs.append(content[i+1] - content[i])
ones = diffs.count(1) + 1
threes = diffs.count(3) + 1
print(ones * threes)