Files
AoC-2020/day10/day10.py
2020-12-10 22:51:27 +01:00

27 lines
494 B
Python

def part1():
file = open("input.txt")
lines = [int(i) for i in file.readlines()]
lines = sorted(lines)
diffs = {1: 0, 3: 0}
diffs[lines[0]] += 1
lines.append(max(lines) + 3)
for i in range(1, len(lines)):
diff = lines[i] - lines[i - 1]
diffs[diff] += 1
print("Part 1 %d " % (diffs[1] * diffs[3]))
def part2():
file = open("input.txt")
lines = [int(i) for i in file.readlines()]
if __name__ == "__main__":
part1()
part2()