1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
#!/usr/bin/env python3
from collections import defaultdict
# START PART 2
from itertools import count
# END PART 2
from typing import DefaultDict, Literal
def main() -> None:
with open("input", "r", encoding="utf-8") as f:
banks = tuple(map(int, f.read().split()))
# START PART 1
memory: DefaultDict[tuple[int, ...], Literal[0]] = defaultdict(int)
# END PART 1 START PART 2
inc = count(start=0, step=1)
memory: DefaultDict[tuple[int, ...], int] = defaultdict(lambda: next(inc))
# END PART 2
while banks not in memory:
memory[banks]
i, v = max(enumerate(banks), key=lambda k: (k[1], -k[0]))
banks = banks[:i] + (0,) + banks[i + 1 :]
for _ in range(v):
i = (i + 1) % 16
banks = banks[:i] + ((banks[i] + 1),) + banks[i + 1 :]
# START PART 1
print(len(memory))
# END PART 1 START PART 2
print(len(memory) - memory[banks])
# END PART 2
if __name__ == "__main__":
main()
|
