37 lines
918 B
Python
37 lines
918 B
Python
from tools.aoc import AOCDay
|
|
from typing import Any
|
|
|
|
class Day(AOCDay):
|
|
inputs = [
|
|
[
|
|
("11111000111110000", "input16"),
|
|
],
|
|
[
|
|
("10111100110110100", "input16"),
|
|
]
|
|
]
|
|
|
|
def get_checksum(self, disk_size: int) -> str:
|
|
state = self.getInput()
|
|
while len(state) < disk_size:
|
|
state = state + "0" + "".join("1" if x == "0" else "0" for x in reversed(state))
|
|
|
|
state = state[:disk_size]
|
|
while len(state) % 2 == 0:
|
|
state = "".join("1" if state[i] == state[i + 1] else "0" for i in range(0, len(state) - 1, 2))
|
|
|
|
return state
|
|
|
|
def part1(self) -> Any:
|
|
disk_size = 272
|
|
return self.get_checksum(disk_size)
|
|
|
|
def part2(self) -> Any:
|
|
disk_size = 35651584
|
|
return self.get_checksum(disk_size)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
day = Day(2016, 16)
|
|
day.run(verbose=True)
|