63 lines
1.5 KiB
Python
63 lines
1.5 KiB
Python
from tools.aoc import AOCDay
|
|
from typing import Any
|
|
|
|
|
|
def cycle(sequence: list, sizes: list, index: int = 0, skip_length: int = 0) -> (list, int, int):
|
|
for size in sizes:
|
|
for inc in range(size // 2):
|
|
l, r = (index + inc) % 256, (index + size - inc - 1) % 256
|
|
sequence[l], sequence[r] = sequence[r], sequence[l]
|
|
index = (index + size + skip_length) % 256
|
|
skip_length += 1
|
|
|
|
return sequence, index, skip_length
|
|
|
|
|
|
def knot_hash(key: str, rounds: int = 64) -> str:
|
|
index = 0
|
|
skip_length = 0
|
|
sequence = list(range(256))
|
|
key = [ord(x) for x in key]
|
|
key += [17, 31, 73, 47, 23]
|
|
|
|
for i in range(rounds):
|
|
sequence, index, skip_length = cycle(sequence, key, index, skip_length)
|
|
|
|
dense = []
|
|
for m in range(16):
|
|
xor = 0
|
|
for i in range(16):
|
|
xor ^= sequence[m * 16 + i]
|
|
|
|
dense.append(xor)
|
|
|
|
hex_string = ""
|
|
for v in dense:
|
|
hex_string += ("%x" % v).rjust(2, '0')
|
|
|
|
return hex_string
|
|
|
|
|
|
class Day(AOCDay):
|
|
inputs = [
|
|
[
|
|
(62238, "input10")
|
|
],
|
|
[
|
|
("2b0c9cc0449507a0db3babd57ad9e8d8", "input10")
|
|
]
|
|
]
|
|
|
|
def part1(self) -> Any:
|
|
sizes = self.getInputAsArraySplit(",", int)
|
|
sequence, _, _ = cycle(list(range(256)), sizes)
|
|
return sequence[0] * sequence[1]
|
|
|
|
def part2(self) -> Any:
|
|
return knot_hash(self.getInput().strip())
|
|
|
|
|
|
if __name__ == '__main__':
|
|
day = Day(2017, 10)
|
|
day.run(verbose=True)
|