mirror of
https://github.com/Crocmagnon/advent-of-code.git
synced 2024-11-14 10:43:58 +01:00
125 lines
3.4 KiB
Python
125 lines
3.4 KiB
Python
from __future__ import annotations
|
|
|
|
import copy
|
|
import dataclasses
|
|
import re
|
|
|
|
|
|
def main(filename: str, expected_part_1: str = None, expected_part_2: str = None):
|
|
print(f"\n+ Running on {filename}")
|
|
with open(filename) as f:
|
|
data = f.read().rstrip().split("\n\n")
|
|
|
|
data_1, data_2 = parse_data(data)
|
|
solution_part_1 = solve_part_1(data_1)
|
|
|
|
print(f"1. Found {solution_part_1}")
|
|
if expected_part_1:
|
|
assert expected_part_1 == solution_part_1
|
|
|
|
solution_part_2 = solve_part_2(data_2)
|
|
print(f"2. Found {solution_part_2}")
|
|
if expected_part_2:
|
|
assert expected_part_2 == solution_part_2
|
|
|
|
|
|
@dataclasses.dataclass
|
|
class Instruction:
|
|
quantity: int
|
|
_source: int
|
|
_destination: int
|
|
|
|
@classmethod
|
|
def from_text(cls, text) -> Instruction:
|
|
match = re.match(r"move (\d+) from (\d+) to (\d+)", text)
|
|
quantity, source, destination = match.groups()
|
|
return cls(int(quantity), int(source), int(destination))
|
|
|
|
@property
|
|
def source(self) -> int:
|
|
return self._source - 1
|
|
|
|
@property
|
|
def destination(self) -> int:
|
|
return self._destination - 1
|
|
|
|
|
|
@dataclasses.dataclass
|
|
class Game:
|
|
stacks: list[list[str]]
|
|
instructions: list[Instruction]
|
|
|
|
def __str__(self):
|
|
return "\n".join(map(str, self.stacks))
|
|
|
|
def play_instructions(self) -> None:
|
|
for instruction in self.instructions:
|
|
self.execute(instruction)
|
|
|
|
def execute(self, instruction: Instruction) -> None:
|
|
raise NotImplementedError()
|
|
|
|
def message(self) -> str:
|
|
msg = ""
|
|
for stack in self.stacks:
|
|
if stack:
|
|
msg += stack[-1]
|
|
return msg
|
|
|
|
|
|
class Game1(Game):
|
|
def execute(self, instruction: Instruction) -> None:
|
|
for _ in range(instruction.quantity):
|
|
item = self.stacks[instruction.source].pop()
|
|
self.stacks[instruction.destination].append(item)
|
|
|
|
|
|
class Game2(Game):
|
|
def execute(self, instruction: Instruction) -> None:
|
|
temp_stack = []
|
|
for _ in range(instruction.quantity):
|
|
item = self.stacks[instruction.source].pop()
|
|
temp_stack.append(item)
|
|
for _ in range(instruction.quantity):
|
|
item = temp_stack.pop()
|
|
self.stacks[instruction.destination].append(item)
|
|
|
|
|
|
DataType = Game
|
|
|
|
|
|
def parse_data(data: list[str]) -> tuple[DataType, DataType]:
|
|
stacks, instructions = data
|
|
parsed_instructions = []
|
|
for instruction in instructions.split("\n"):
|
|
parsed_instructions.append(Instruction.from_text(instruction))
|
|
stacks = list(stacks.split("\n")[::-1])
|
|
header = stacks[0]
|
|
parsed_stacks = []
|
|
for index, char in enumerate(header):
|
|
if char != " ":
|
|
stack = []
|
|
for row in stacks[1:]:
|
|
char = row[index]
|
|
if char == " ":
|
|
break
|
|
stack.append(char)
|
|
parsed_stacks.append(stack)
|
|
return Game1(
|
|
copy.deepcopy(parsed_stacks), copy.deepcopy(parsed_instructions)
|
|
), Game2(copy.deepcopy(parsed_stacks), copy.deepcopy(parsed_instructions))
|
|
|
|
|
|
def solve_part_1(data: DataType) -> str:
|
|
data.play_instructions()
|
|
return data.message()
|
|
|
|
|
|
def solve_part_2(data: DataType) -> str:
|
|
data.play_instructions()
|
|
return data.message()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main("inputs/day05-test1", expected_part_1="CMZ", expected_part_2="MCD")
|
|
main("inputs/day05", expected_part_1="BWNCQRMDB", expected_part_2="NHWZCBNBF")
|