Solve day 5 part 1

.pre-commit-config.yaml

@@ -16,6 +16,7 @@ repos:
       - id: trailing-whitespace
         args:
           - --markdown-linebreak-ext=md
+        exclude: ^2022/inputs/day05
   - repo: https://github.com/asottile/pyupgrade
     rev: v3.2.2
     hooks:

2022/day05_supply_stacks.py

@@ -0,0 +1,105 @@
+from __future__ import annotations
+
+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 = parse_data(data)
+    solution_part_1 = solve_part_1(data)
+
+    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)
+    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 play_instructions(self) -> None:
+        for instruction in self.instructions:
+            self.execute(instruction)
+
+    def execute(self, instruction: Instruction) -> None:
+        for _ in range(instruction.quantity):
+            item = self.stacks[instruction.source].pop()
+            self.stacks[instruction.destination].append(item)
+
+    def message(self) -> str:
+        msg = ""
+        for stack in self.stacks:
+            if stack:
+                msg += stack[-1]
+        return msg
+
+    def __str__(self):
+        return "\n".join(map(str, self.stacks))
+
+
+DataType = Game
+
+
+def parse_data(data: list[str]) -> 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 Game(parsed_stacks, parsed_instructions)
+
+
+def solve_part_1(data: DataType) -> str:
+    data.play_instructions()
+    return data.message()
+
+
+def solve_part_2(data: DataType) -> str:
+    return "0"
+
+
+if __name__ == "__main__":
+    main("inputs/day05-test1", expected_part_1="CMZ")
+    main("inputs/day05")

2022/inputs/day05

@@ -0,0 +1,514 @@
+                    [Q]     [P] [P]
+                [G] [V] [S] [Z] [F]
+            [W] [V] [F] [Z] [W] [Q]
+        [V] [T] [N] [J] [W] [B] [W]
+    [Z] [L] [V] [B] [C] [R] [N] [M]
+[C] [W] [R] [H] [H] [P] [T] [M] [B]
+[Q] [Q] [M] [Z] [Z] [N] [G] [G] [J]
+[B] [R] [B] [C] [D] [H] [D] [C] [N]
+ 1   2   3   4   5   6   7   8   9 
+
+move 3 from 6 to 2
+move 5 from 6 to 7
+move 6 from 2 to 5
+move 1 from 9 to 7
+move 1 from 1 to 9
+move 1 from 5 to 3
+move 1 from 2 to 5
+move 3 from 4 to 5
+move 10 from 7 to 3
+move 1 from 4 to 9
+move 6 from 8 to 7
+move 4 from 7 to 8
+move 1 from 7 to 3
+move 1 from 1 to 2
+move 1 from 2 to 8
+move 1 from 9 to 1
+move 3 from 9 to 4
+move 4 from 8 to 3
+move 4 from 7 to 1
+move 4 from 4 to 6
+move 2 from 8 to 7
+move 9 from 3 to 8
+move 2 from 7 to 4
+move 3 from 4 to 9
+move 4 from 1 to 9
+move 4 from 3 to 9
+move 2 from 1 to 4
+move 1 from 4 to 6
+move 3 from 3 to 2
+move 1 from 2 to 8
+move 1 from 2 to 7
+move 3 from 6 to 2
+move 2 from 6 to 7
+move 4 from 2 to 3
+move 3 from 7 to 9
+move 2 from 5 to 6
+move 15 from 9 to 4
+move 4 from 9 to 2
+move 12 from 5 to 4
+move 9 from 8 to 5
+move 25 from 4 to 7
+move 1 from 4 to 7
+move 1 from 4 to 8
+move 2 from 2 to 5
+move 1 from 4 to 2
+move 23 from 7 to 6
+move 2 from 5 to 2
+move 22 from 6 to 8
+move 4 from 5 to 9
+move 1 from 7 to 9
+move 2 from 6 to 4
+move 2 from 4 to 7
+move 25 from 8 to 3
+move 1 from 2 to 1
+move 3 from 2 to 3
+move 1 from 6 to 8
+move 1 from 1 to 8
+move 1 from 2 to 8
+move 1 from 8 to 1
+move 4 from 5 to 7
+move 1 from 8 to 4
+move 5 from 9 to 8
+move 5 from 8 to 9
+move 1 from 8 to 5
+move 3 from 5 to 4
+move 3 from 9 to 1
+move 30 from 3 to 4
+move 3 from 1 to 4
+move 2 from 9 to 5
+move 4 from 7 to 9
+move 16 from 4 to 8
+move 6 from 3 to 9
+move 3 from 7 to 3
+move 19 from 4 to 7
+move 8 from 9 to 4
+move 1 from 1 to 9
+move 13 from 7 to 9
+move 3 from 7 to 8
+move 3 from 5 to 9
+move 4 from 8 to 3
+move 2 from 7 to 3
+move 14 from 9 to 4
+move 10 from 3 to 1
+move 12 from 4 to 8
+move 6 from 1 to 9
+move 1 from 1 to 2
+move 1 from 7 to 1
+move 6 from 9 to 3
+move 17 from 8 to 6
+move 10 from 8 to 5
+move 1 from 7 to 8
+move 1 from 9 to 5
+move 2 from 3 to 1
+move 4 from 5 to 9
+move 1 from 8 to 7
+move 6 from 9 to 7
+move 4 from 4 to 2
+move 3 from 4 to 6
+move 4 from 5 to 9
+move 4 from 9 to 3
+move 1 from 2 to 4
+move 4 from 4 to 7
+move 3 from 5 to 3
+move 1 from 4 to 5
+move 5 from 1 to 2
+move 1 from 1 to 9
+move 7 from 2 to 7
+move 1 from 5 to 7
+move 8 from 3 to 5
+move 20 from 6 to 7
+move 9 from 7 to 9
+move 2 from 2 to 9
+move 2 from 3 to 1
+move 2 from 1 to 3
+move 2 from 3 to 4
+move 2 from 4 to 6
+move 1 from 3 to 9
+move 1 from 4 to 9
+move 1 from 6 to 9
+move 2 from 5 to 8
+move 2 from 8 to 5
+move 1 from 6 to 7
+move 2 from 5 to 8
+move 6 from 9 to 5
+move 2 from 8 to 6
+move 11 from 9 to 2
+move 1 from 6 to 5
+move 11 from 2 to 5
+move 1 from 6 to 4
+move 7 from 5 to 9
+move 7 from 9 to 1
+move 1 from 4 to 9
+move 28 from 7 to 5
+move 1 from 7 to 5
+move 5 from 5 to 9
+move 5 from 9 to 3
+move 6 from 1 to 8
+move 1 from 1 to 7
+move 5 from 3 to 2
+move 1 from 7 to 8
+move 7 from 8 to 1
+move 1 from 9 to 4
+move 2 from 2 to 5
+move 22 from 5 to 3
+move 1 from 7 to 8
+move 1 from 4 to 7
+move 1 from 8 to 9
+move 1 from 9 to 4
+move 14 from 5 to 7
+move 5 from 5 to 9
+move 19 from 3 to 4
+move 1 from 2 to 9
+move 2 from 2 to 5
+move 1 from 5 to 1
+move 6 from 1 to 7
+move 2 from 7 to 6
+move 1 from 1 to 9
+move 2 from 5 to 8
+move 8 from 4 to 5
+move 3 from 4 to 7
+move 3 from 3 to 5
+move 2 from 8 to 9
+move 16 from 7 to 5
+move 9 from 4 to 6
+move 22 from 5 to 3
+move 1 from 5 to 8
+move 1 from 8 to 7
+move 10 from 3 to 4
+move 1 from 5 to 4
+move 10 from 4 to 5
+move 8 from 5 to 2
+move 5 from 2 to 7
+move 5 from 7 to 1
+move 4 from 7 to 6
+move 3 from 9 to 7
+move 2 from 2 to 3
+move 3 from 5 to 1
+move 6 from 9 to 7
+move 5 from 7 to 8
+move 6 from 1 to 5
+move 6 from 3 to 4
+move 4 from 4 to 2
+move 1 from 4 to 6
+move 5 from 8 to 7
+move 3 from 2 to 3
+move 1 from 1 to 4
+move 1 from 1 to 9
+move 2 from 2 to 1
+move 2 from 4 to 3
+move 4 from 3 to 7
+move 3 from 7 to 3
+move 13 from 6 to 1
+move 1 from 9 to 2
+move 6 from 3 to 5
+move 8 from 1 to 4
+move 1 from 2 to 7
+move 9 from 4 to 9
+move 7 from 5 to 1
+move 2 from 5 to 6
+move 1 from 1 to 4
+move 1 from 4 to 3
+move 2 from 1 to 2
+move 5 from 3 to 6
+move 2 from 6 to 1
+move 13 from 7 to 6
+move 2 from 3 to 4
+move 2 from 2 to 9
+move 2 from 7 to 8
+move 6 from 9 to 2
+move 1 from 9 to 3
+move 1 from 5 to 2
+move 7 from 1 to 2
+move 1 from 6 to 7
+move 1 from 4 to 8
+move 1 from 3 to 1
+move 1 from 7 to 8
+move 7 from 1 to 9
+move 4 from 8 to 6
+move 1 from 5 to 3
+move 9 from 9 to 5
+move 1 from 1 to 2
+move 14 from 2 to 7
+move 2 from 9 to 3
+move 13 from 5 to 3
+move 24 from 6 to 9
+move 6 from 3 to 5
+move 14 from 7 to 9
+move 1 from 4 to 1
+move 20 from 9 to 7
+move 9 from 3 to 8
+move 15 from 9 to 6
+move 1 from 5 to 8
+move 1 from 2 to 3
+move 14 from 6 to 3
+move 2 from 3 to 4
+move 2 from 3 to 6
+move 13 from 7 to 1
+move 8 from 3 to 5
+move 1 from 3 to 9
+move 8 from 5 to 4
+move 4 from 5 to 2
+move 10 from 1 to 3
+move 6 from 4 to 5
+move 4 from 5 to 1
+move 3 from 1 to 6
+move 7 from 8 to 2
+move 4 from 4 to 3
+move 13 from 3 to 6
+move 3 from 8 to 1
+move 3 from 7 to 8
+move 3 from 8 to 4
+move 1 from 4 to 2
+move 2 from 3 to 4
+move 1 from 5 to 7
+move 4 from 7 to 1
+move 2 from 3 to 5
+move 3 from 2 to 1
+move 1 from 4 to 7
+move 7 from 2 to 4
+move 2 from 4 to 3
+move 1 from 7 to 5
+move 4 from 9 to 5
+move 1 from 4 to 2
+move 3 from 2 to 9
+move 8 from 1 to 7
+move 1 from 3 to 5
+move 7 from 5 to 7
+move 10 from 6 to 4
+move 1 from 5 to 1
+move 4 from 1 to 3
+move 9 from 7 to 6
+move 3 from 1 to 8
+move 12 from 4 to 6
+move 5 from 4 to 6
+move 2 from 9 to 3
+move 3 from 8 to 7
+move 1 from 1 to 3
+move 3 from 7 to 8
+move 5 from 7 to 5
+move 1 from 7 to 5
+move 2 from 3 to 1
+move 2 from 8 to 7
+move 3 from 5 to 1
+move 1 from 9 to 7
+move 1 from 8 to 3
+move 4 from 7 to 8
+move 4 from 5 to 9
+move 4 from 1 to 7
+move 3 from 8 to 6
+move 1 from 8 to 1
+move 1 from 7 to 1
+move 1 from 5 to 8
+move 1 from 8 to 7
+move 7 from 3 to 1
+move 3 from 9 to 1
+move 1 from 9 to 3
+move 28 from 6 to 3
+move 3 from 7 to 8
+move 2 from 8 to 2
+move 1 from 2 to 7
+move 2 from 6 to 1
+move 18 from 3 to 9
+move 5 from 3 to 4
+move 2 from 7 to 4
+move 2 from 1 to 8
+move 1 from 2 to 6
+move 7 from 6 to 4
+move 4 from 4 to 3
+move 3 from 8 to 1
+move 4 from 9 to 8
+move 1 from 4 to 8
+move 9 from 1 to 6
+move 5 from 1 to 3
+move 4 from 6 to 7
+move 7 from 6 to 3
+move 5 from 8 to 1
+move 12 from 3 to 6
+move 7 from 6 to 4
+move 4 from 3 to 5
+move 5 from 6 to 7
+move 12 from 4 to 3
+move 6 from 1 to 4
+move 4 from 4 to 2
+move 14 from 9 to 8
+move 17 from 3 to 2
+move 5 from 4 to 9
+move 1 from 9 to 6
+move 5 from 2 to 1
+move 1 from 9 to 8
+move 5 from 1 to 6
+move 2 from 2 to 6
+move 12 from 2 to 4
+move 6 from 7 to 2
+move 3 from 7 to 6
+move 3 from 9 to 8
+move 5 from 4 to 7
+move 4 from 2 to 6
+move 3 from 6 to 8
+move 5 from 8 to 2
+move 7 from 6 to 8
+move 1 from 7 to 3
+move 6 from 4 to 3
+move 1 from 8 to 1
+move 1 from 5 to 7
+move 2 from 6 to 8
+move 13 from 8 to 2
+move 3 from 5 to 4
+move 1 from 1 to 2
+move 3 from 6 to 2
+move 1 from 1 to 4
+move 4 from 4 to 8
+move 8 from 3 to 1
+move 2 from 4 to 8
+move 15 from 2 to 4
+move 16 from 8 to 3
+move 1 from 8 to 6
+move 1 from 7 to 2
+move 8 from 1 to 2
+move 1 from 6 to 8
+move 6 from 3 to 1
+move 3 from 3 to 8
+move 6 from 3 to 1
+move 6 from 2 to 9
+move 2 from 1 to 4
+move 1 from 8 to 5
+move 8 from 2 to 9
+move 8 from 1 to 4
+move 3 from 8 to 6
+move 21 from 4 to 7
+move 1 from 9 to 7
+move 2 from 6 to 8
+move 1 from 5 to 1
+move 1 from 3 to 9
+move 8 from 9 to 4
+move 1 from 1 to 7
+move 1 from 1 to 4
+move 1 from 6 to 8
+move 1 from 9 to 3
+move 2 from 9 to 5
+move 2 from 5 to 3
+move 1 from 9 to 4
+move 3 from 8 to 2
+move 1 from 1 to 4
+move 4 from 4 to 9
+move 3 from 3 to 2
+move 5 from 9 to 1
+move 17 from 7 to 1
+move 1 from 9 to 1
+move 2 from 2 to 4
+move 1 from 4 to 2
+move 8 from 2 to 9
+move 5 from 4 to 5
+move 6 from 4 to 8
+move 20 from 1 to 6
+move 2 from 9 to 8
+move 1 from 2 to 9
+move 2 from 8 to 7
+move 8 from 7 to 8
+move 4 from 5 to 9
+move 14 from 8 to 7
+move 1 from 5 to 7
+move 7 from 9 to 1
+move 3 from 6 to 4
+move 3 from 9 to 7
+move 12 from 6 to 7
+move 22 from 7 to 9
+move 2 from 2 to 5
+move 10 from 1 to 7
+move 1 from 4 to 1
+move 2 from 6 to 1
+move 1 from 1 to 3
+move 2 from 4 to 8
+move 2 from 8 to 6
+move 1 from 3 to 8
+move 1 from 4 to 1
+move 2 from 5 to 3
+move 1 from 8 to 4
+move 2 from 3 to 7
+move 19 from 9 to 7
+move 1 from 1 to 4
+move 2 from 9 to 1
+move 2 from 1 to 6
+move 1 from 6 to 5
+move 42 from 7 to 8
+move 1 from 7 to 6
+move 2 from 4 to 8
+move 7 from 6 to 8
+move 2 from 1 to 5
+move 2 from 9 to 5
+move 14 from 8 to 3
+move 22 from 8 to 2
+move 3 from 5 to 6
+move 10 from 8 to 6
+move 5 from 8 to 9
+move 12 from 6 to 7
+move 2 from 5 to 1
+move 5 from 3 to 2
+move 7 from 3 to 5
+move 2 from 5 to 1
+move 2 from 3 to 7
+move 4 from 1 to 2
+move 1 from 5 to 7
+move 1 from 5 to 4
+move 1 from 6 to 2
+move 1 from 9 to 2
+move 9 from 7 to 3
+move 1 from 4 to 1
+move 3 from 7 to 5
+move 4 from 3 to 2
+move 5 from 2 to 3
+move 2 from 5 to 2
+move 34 from 2 to 9
+move 1 from 1 to 5
+move 15 from 9 to 3
+move 2 from 3 to 2
+move 1 from 5 to 4
+move 7 from 3 to 8
+move 3 from 9 to 2
+move 6 from 9 to 4
+move 5 from 9 to 3
+move 4 from 4 to 6
+move 1 from 6 to 8
+move 1 from 3 to 5
+move 6 from 3 to 2
+move 1 from 4 to 9
+move 2 from 4 to 2
+move 4 from 5 to 8
+move 1 from 5 to 6
+move 1 from 7 to 6
+move 1 from 9 to 6
+move 1 from 7 to 2
+move 12 from 8 to 7
+move 2 from 7 to 3
+move 4 from 6 to 9
+move 7 from 9 to 4
+move 9 from 3 to 9
+move 11 from 7 to 4
+move 3 from 9 to 6
+move 1 from 4 to 1
+move 15 from 4 to 3
+move 2 from 4 to 1
+move 3 from 1 to 4
+move 17 from 3 to 7
+move 4 from 3 to 7
+move 7 from 9 to 2
+move 3 from 4 to 1
+move 4 from 6 to 9
+move 1 from 9 to 6
+move 1 from 3 to 1
+move 5 from 7 to 9
+move 8 from 9 to 4
+move 1 from 1 to 6
+move 6 from 4 to 9
+move 4 from 2 to 3
+move 1 from 4 to 3
+move 1 from 4 to 9
+move 1 from 1 to 7
+move 1 from 7 to 9
+move 3 from 6 to 2
+move 9 from 2 to 3
+move 1 from 9 to 4
+move 1 from 1 to 5
+move 12 from 7 to 6
+move 4 from 9 to 8

2022/inputs/day05-test1

@@ -0,0 +1,9 @@
+    [D]    
+[N] [C]    
+[Z] [M] [P]
+ 1   2   3 
+
+move 1 from 2 to 1
+move 3 from 1 to 3
+move 2 from 2 to 1
+move 1 from 1 to 2