diff options
Diffstat (limited to 'day5')
| -rw-r--r-- | day5/__init__.py | 108 | ||||
| -rw-r--r-- | day5/example.txt | 9 | ||||
| -rw-r--r-- | day5/input.txt | 512 |
3 files changed, 629 insertions, 0 deletions
diff --git a/day5/__init__.py b/day5/__init__.py new file mode 100644 index 0000000..a9aaba5 --- /dev/null +++ b/day5/__init__.py | |||
| @@ -0,0 +1,108 @@ | |||
| 1 | # -*- coding: utf-8 -*- | ||
| 2 | import re | ||
| 3 | from abc import ABC | ||
| 4 | from typing import Iterator, Any, TypedDict, Tuple, List | ||
| 5 | from queue import LifoQueue | ||
| 6 | |||
| 7 | from aoc import BaseAssignment | ||
| 8 | |||
| 9 | move_regex = re.compile("^move (?P<amount>\d+) from (?P<start>\d+) to (?P<end>\d+)$") | ||
| 10 | |||
| 11 | |||
| 12 | class Move(TypedDict): | ||
| 13 | amount: int | ||
| 14 | start: int | ||
| 15 | end: int | ||
| 16 | |||
| 17 | |||
| 18 | class Assignment(BaseAssignment, ABC): | ||
| 19 | def parse_move(self, move: str) -> Move: | ||
| 20 | match = move_regex.match(move) | ||
| 21 | if match: | ||
| 22 | move = match.groupdict() | ||
| 23 | return Move( | ||
| 24 | amount=int(move["amount"]), | ||
| 25 | start=int(move["start"]), | ||
| 26 | end=int(move["end"]), | ||
| 27 | ) | ||
| 28 | |||
| 29 | def parse_stacks(self, string_stacks: List[str]) -> List[LifoQueue]: | ||
| 30 | stacks = [] | ||
| 31 | columns = string_stacks.pop().replace(" ", "") | ||
| 32 | |||
| 33 | for i in range(len(columns)): | ||
| 34 | stacks.append(LifoQueue()) | ||
| 35 | |||
| 36 | for line in reversed(string_stacks): | ||
| 37 | for index, stack in enumerate(stacks): | ||
| 38 | item_index = (index * 4) + 1 | ||
| 39 | |||
| 40 | try: | ||
| 41 | item = line[item_index] | ||
| 42 | if item != " ": | ||
| 43 | stack.put(item) | ||
| 44 | except IndexError: | ||
| 45 | pass | ||
| 46 | |||
| 47 | return stacks | ||
| 48 | |||
| 49 | def parse_input(self, input: Iterator) -> Tuple[List[LifoQueue], List[Move]]: | ||
| 50 | parsing_stacks = True | ||
| 51 | tmp_stacks = [] | ||
| 52 | stacks = [] | ||
| 53 | moves = [] | ||
| 54 | |||
| 55 | for line in input: | ||
| 56 | if line == "": | ||
| 57 | parsing_stacks = False | ||
| 58 | stacks = self.parse_stacks(tmp_stacks) | ||
| 59 | continue | ||
| 60 | |||
| 61 | if parsing_stacks: | ||
| 62 | tmp_stacks.append(line) | ||
| 63 | else: | ||
| 64 | moves.append(self.parse_move(line)) | ||
| 65 | |||
| 66 | return stacks, moves | ||
| 67 | |||
| 68 | def execute_moves(self, stacks: List[LifoQueue], moves: List[Move]): | ||
| 69 | raise NotImplementedError() | ||
| 70 | |||
| 71 | def run(self, input: Iterator) -> Any: | ||
| 72 | stacks, moves = self.parse_input(input) | ||
| 73 | |||
| 74 | self.execute_moves(stacks, moves) | ||
| 75 | |||
| 76 | return "".join([stack.get() for stack in stacks]) | ||
| 77 | |||
| 78 | |||
| 79 | class AssignmentOne(Assignment): | ||
| 80 | example_result = "CMZ" | ||
| 81 | |||
| 82 | def execute_moves(self, stacks: List[LifoQueue], moves: List[Move]): | ||
| 83 | for move in moves: | ||
| 84 | start = stacks[move["start"] - 1] | ||
| 85 | end = stacks[move["end"] - 1] | ||
| 86 | |||
| 87 | for _ in range(move["amount"]): | ||
| 88 | end.put(start.get()) | ||
| 89 | |||
| 90 | return stacks | ||
| 91 | |||
| 92 | |||
| 93 | class AssignmentTwo(Assignment): | ||
| 94 | example_result = "MCD" | ||
| 95 | |||
| 96 | def execute_moves(self, stacks: List[LifoQueue], moves: List[Move]): | ||
| 97 | for move in moves: | ||
| 98 | start = stacks[move["start"] - 1] | ||
| 99 | end = stacks[move["end"] - 1] | ||
| 100 | |||
| 101 | tmp_stack = LifoQueue() | ||
| 102 | for _ in range(move["amount"]): | ||
| 103 | tmp_stack.put(start.get()) | ||
| 104 | |||
| 105 | while not tmp_stack.empty(): | ||
| 106 | end.put(tmp_stack.get()) | ||
| 107 | |||
| 108 | return stacks | ||
diff --git a/day5/example.txt b/day5/example.txt new file mode 100644 index 0000000..42ef47f --- /dev/null +++ b/day5/example.txt | |||
| @@ -0,0 +1,9 @@ | |||
| 1 | [D] | ||
| 2 | [N] [C] | ||
| 3 | [Z] [M] [P] | ||
| 4 | 1 2 3 | ||
| 5 | |||
| 6 | move 1 from 2 to 1 | ||
| 7 | move 3 from 1 to 3 | ||
| 8 | move 2 from 2 to 1 | ||
| 9 | move 1 from 1 to 2 | ||
diff --git a/day5/input.txt b/day5/input.txt new file mode 100644 index 0000000..54a285f --- /dev/null +++ b/day5/input.txt | |||
| @@ -0,0 +1,512 @@ | |||
| 1 | [F] [L] [M] | ||
| 2 | [T] [H] [V] [G] [V] | ||
| 3 | [N] [T] [D] [R] [N] [D] | ||
| 4 | [Z] [B] [C] [P] [B] [R] [Z] | ||
| 5 | [M] [J] [N] [M] [F] [M] [V] [H] | ||
| 6 | [G] [J] [L] [J] [S] [C] [G] [M] [F] | ||
| 7 | [H] [W] [V] [P] [W] [H] [H] [N] [N] | ||
| 8 | [J] [V] [G] [B] [F] [G] [D] [H] [G] | ||
| 9 | 1 2 3 4 5 6 7 8 9 | ||
| 10 | |||
| 11 | move 6 from 4 to 3 | ||
| 12 | move 5 from 8 to 9 | ||
| 13 | move 1 from 4 to 5 | ||
| 14 | move 1 from 4 to 5 | ||
| 15 | move 2 from 2 to 7 | ||
| 16 | move 2 from 1 to 6 | ||
| 17 | move 9 from 6 to 1 | ||
| 18 | move 12 from 3 to 5 | ||
| 19 | move 1 from 8 to 4 | ||
| 20 | move 3 from 1 to 5 | ||
| 21 | move 1 from 6 to 7 | ||
| 22 | move 10 from 5 to 2 | ||
| 23 | move 14 from 5 to 1 | ||
| 24 | move 8 from 7 to 9 | ||
| 25 | move 11 from 2 to 9 | ||
| 26 | move 1 from 3 to 9 | ||
| 27 | move 11 from 1 to 5 | ||
| 28 | move 2 from 1 to 9 | ||
| 29 | move 1 from 4 to 8 | ||
| 30 | move 6 from 1 to 5 | ||
| 31 | move 1 from 8 to 3 | ||
| 32 | move 16 from 5 to 1 | ||
| 33 | move 4 from 1 to 3 | ||
| 34 | move 1 from 5 to 6 | ||
| 35 | move 4 from 3 to 4 | ||
| 36 | move 1 from 6 to 7 | ||
| 37 | move 21 from 9 to 6 | ||
| 38 | move 2 from 1 to 9 | ||
| 39 | move 2 from 4 to 9 | ||
| 40 | move 5 from 9 to 4 | ||
| 41 | move 9 from 1 to 6 | ||
| 42 | move 6 from 4 to 6 | ||
| 43 | move 1 from 6 to 2 | ||
| 44 | move 1 from 7 to 6 | ||
| 45 | move 1 from 3 to 2 | ||
| 46 | move 8 from 6 to 9 | ||
| 47 | move 3 from 1 to 8 | ||
| 48 | move 1 from 2 to 1 | ||
| 49 | move 13 from 6 to 3 | ||
| 50 | move 1 from 1 to 9 | ||
| 51 | move 2 from 1 to 6 | ||
| 52 | move 3 from 8 to 4 | ||
| 53 | move 4 from 4 to 9 | ||
| 54 | move 3 from 1 to 3 | ||
| 55 | move 22 from 9 to 8 | ||
| 56 | move 1 from 2 to 9 | ||
| 57 | move 6 from 8 to 9 | ||
| 58 | move 15 from 6 to 5 | ||
| 59 | move 5 from 8 to 9 | ||
| 60 | move 11 from 9 to 8 | ||
| 61 | move 13 from 5 to 1 | ||
| 62 | move 1 from 6 to 5 | ||
| 63 | move 1 from 9 to 3 | ||
| 64 | move 21 from 8 to 3 | ||
| 65 | move 3 from 5 to 3 | ||
| 66 | move 11 from 1 to 2 | ||
| 67 | move 25 from 3 to 1 | ||
| 68 | move 5 from 1 to 7 | ||
| 69 | move 20 from 1 to 7 | ||
| 70 | move 1 from 6 to 7 | ||
| 71 | move 16 from 3 to 9 | ||
| 72 | move 8 from 9 to 6 | ||
| 73 | move 1 from 1 to 5 | ||
| 74 | move 5 from 9 to 4 | ||
| 75 | move 2 from 2 to 1 | ||
| 76 | move 2 from 9 to 4 | ||
| 77 | move 1 from 9 to 4 | ||
| 78 | move 1 from 8 to 4 | ||
| 79 | move 1 from 5 to 2 | ||
| 80 | move 3 from 4 to 6 | ||
| 81 | move 1 from 4 to 7 | ||
| 82 | move 9 from 7 to 6 | ||
| 83 | move 5 from 4 to 6 | ||
| 84 | move 7 from 7 to 2 | ||
| 85 | move 1 from 1 to 6 | ||
| 86 | move 11 from 2 to 5 | ||
| 87 | move 10 from 5 to 1 | ||
| 88 | move 1 from 6 to 8 | ||
| 89 | move 1 from 5 to 7 | ||
| 90 | move 24 from 6 to 1 | ||
| 91 | move 12 from 1 to 4 | ||
| 92 | move 12 from 4 to 8 | ||
| 93 | move 2 from 2 to 7 | ||
| 94 | move 3 from 7 to 2 | ||
| 95 | move 5 from 2 to 8 | ||
| 96 | move 9 from 8 to 9 | ||
| 97 | move 9 from 8 to 5 | ||
| 98 | move 1 from 9 to 1 | ||
| 99 | move 14 from 1 to 8 | ||
| 100 | move 11 from 7 to 9 | ||
| 101 | move 4 from 1 to 3 | ||
| 102 | move 7 from 1 to 2 | ||
| 103 | move 3 from 3 to 7 | ||
| 104 | move 12 from 9 to 7 | ||
| 105 | move 8 from 7 to 2 | ||
| 106 | move 4 from 9 to 2 | ||
| 107 | move 1 from 3 to 6 | ||
| 108 | move 5 from 5 to 9 | ||
| 109 | move 14 from 2 to 1 | ||
| 110 | move 8 from 9 to 4 | ||
| 111 | move 6 from 4 to 5 | ||
| 112 | move 5 from 5 to 7 | ||
| 113 | move 1 from 8 to 2 | ||
| 114 | move 2 from 4 to 6 | ||
| 115 | move 4 from 7 to 3 | ||
| 116 | move 10 from 8 to 4 | ||
| 117 | move 2 from 3 to 6 | ||
| 118 | move 7 from 7 to 6 | ||
| 119 | move 10 from 4 to 8 | ||
| 120 | move 5 from 1 to 6 | ||
| 121 | move 8 from 2 to 1 | ||
| 122 | move 7 from 6 to 8 | ||
| 123 | move 9 from 6 to 5 | ||
| 124 | move 16 from 1 to 6 | ||
| 125 | move 2 from 3 to 9 | ||
| 126 | move 1 from 7 to 4 | ||
| 127 | move 2 from 9 to 1 | ||
| 128 | move 14 from 6 to 7 | ||
| 129 | move 1 from 6 to 3 | ||
| 130 | move 2 from 6 to 3 | ||
| 131 | move 9 from 5 to 7 | ||
| 132 | move 3 from 1 to 6 | ||
