| 12345678910111213141516171819202122232425262728293031323334 |
- # Read lines
- lines = []
- with open("10.input") as f:
- for line in f.readlines():
- lines.append(int(line.strip()))
- lines.sort()
- current = 0
- diffs = [0, 0, 0, 0]
- for next in lines:
- diffs[next - current] += 1
- current = next
- diffs[3] += 1 # Device joltage
- print("Answer 1: " + str(diffs[1] * diffs[3]))
- count_from = {}
- count_from[max(lines)] = 1
- lines.reverse()
- lines.pop(0)
- for current in lines + [0]:
- count_from[current] = 0
- for i in range(1, 4):
- count_from[current] += count_from.get(current + i, 0)
- print("Answer 2: " + str(count_from[0]))
|
