Mikaël Capelle
ce315b8778
Co-authored-by: Mikael CAPELLE <mikael.capelle@thalesaleniaspace.com> Co-authored-by: Mikaël Capelle <capelle.mikael@gmail.com> Reviewed-on: #3
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
import copy
|
|
from typing import Any, Iterator
|
|
|
|
from ..base import BaseSolver
|
|
|
|
|
|
class Solver(BaseSolver):
|
|
def solve(self, input: str) -> Iterator[Any]:
|
|
blocks_s, moves_s = (part.splitlines() for part in input.split("\n\n"))
|
|
|
|
blocks: dict[str, list[str]] = {stack: [] for stack in blocks_s[-1].split()}
|
|
|
|
# this codes assumes that the lines are regular, i.e., 4 characters per "crate" in the
|
|
# form of '[X] ' (including the trailing space)
|
|
#
|
|
for block in blocks_s[-2::-1]:
|
|
for stack, index in zip(blocks, range(0, len(block), 4)):
|
|
crate = block[index + 1 : index + 2].strip()
|
|
|
|
if crate:
|
|
blocks[stack].append(crate)
|
|
|
|
# part 1 - deep copy for part 2
|
|
blocks_1 = copy.deepcopy(blocks)
|
|
|
|
for move in moves_s:
|
|
_, count_s, _, from_, _, to_ = move.strip().split()
|
|
|
|
for _i in range(int(count_s)):
|
|
blocks_1[to_].append(blocks_1[from_].pop())
|
|
|
|
# part 2
|
|
blocks_2 = copy.deepcopy(blocks)
|
|
|
|
for move in moves_s:
|
|
_, count_s, _, from_, _, to_ = move.strip().split()
|
|
count = int(count_s)
|
|
|
|
blocks_2[to_].extend(blocks_2[from_][-count:])
|
|
del blocks_2[from_][-count:]
|
|
|
|
yield "".join(s[-1] for s in blocks_1.values())
|
|
yield "".join(s[-1] for s in blocks_2.values())
|