Implement 2019 day 16 part 2

This commit is contained in:
2021-02-02 19:57:24 +01:00
parent 2d561eab7d
commit beae045f55
3 changed files with 49 additions and 15 deletions

View File

@@ -27,7 +27,31 @@ def simulate(numbers: List[int]) -> str:
return ''.join(str(s) for s in numbers[:8])
def simulate2(numbers: List[int]) -> str:
numbers = numpy.tile(numpy.array(numbers, dtype=numpy.int), 10000)
starting_index = 0
for n in numbers[:7]:
starting_index *= 10
starting_index += n
assert starting_index > len(numbers) / 2
numbers = numbers[starting_index:]
for _ in range(100):
numbers = numpy.abs(numpy.flip(numpy.cumsum(numpy.flip(numbers)))) % 10
return ''.join(str(s) for s in numbers[:8])
def part1(data: TextIO) -> str:
numbers = read_input(data)
return simulate(numbers)
def part2(data: TextIO) -> str:
numbers = read_input(data)
return simulate2(numbers)