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