Solve 2022 day 02

puzzles/solutions/2022/d02/consts.py

@@ -0,0 +1,30 @@
+ROCK = "X"
+PAPER = "Y"
+SCISSORS = "Z"
+OPPONENT_ROCK = "A"
+OPPONENT_PAPER = "B"
+OPPONENT_SCISSORS = "C"
+
+IDENTICAL_SHAPES = {
+    OPPONENT_ROCK: ROCK,
+    OPPONENT_PAPER: PAPER,
+    OPPONENT_SCISSORS: SCISSORS,
+}
+
+WINNING_MOVES = {
+    ROCK: SCISSORS,
+    PAPER: ROCK,
+    SCISSORS: PAPER,
+}
+DRAW_MOVES = {shape: shape for shape in (ROCK, PAPER, SCISSORS)}
+LOSING_MOVES = {
+    ROCK: PAPER,
+    PAPER: SCISSORS,
+    SCISSORS: ROCK,
+}
+OUTCOME_TO_MAPPING = {"X": LOSING_MOVES, "Y": DRAW_MOVES, "Z": WINNING_MOVES}
+
+DRAW_SCORE = 3
+WIN_SCORE = 6
+
+SHAPE_TO_SCORE = {ROCK: 1, PAPER: 2, SCISSORS: 3}

puzzles/solutions/2022/d02/p1.py

@@ -0,0 +1,45 @@
+import sys
+
+import consts
+
+
+def get_moves(input_text: str) -> tuple[tuple[str, str], ...]:
+    """
+    :param input_text: puzzle input
+    :return: list of (opponent move, our move) pairs
+    """
+    input_text = input_text.translate(
+        str.maketrans(consts.IDENTICAL_SHAPES)
+    )  # Replace opponent shapes with generic shapes.
+    moves = input_text.splitlines()
+    return tuple(tuple(move.split()) for move in moves)
+
+
+def calculate_move_score(move: tuple[str, str]) -> int:
+    """
+    :param move: pair of (opponent move, our move)
+    :return: total score of the move
+    """
+    score = 0
+    opponent_move, our_move = move
+    if consts.WINNING_MOVES[our_move] == opponent_move:
+        score += consts.WIN_SCORE
+    elif opponent_move == our_move:
+        score += consts.DRAW_SCORE
+
+    score += consts.SHAPE_TO_SCORE[our_move]
+    return score
+
+
+def get_answer(input_text: str):
+    """Return the total score if everything goes exactly according to our strategy guide."""
+    moves = get_moves(input_text)
+    total_score = sum(calculate_move_score(move) for move in moves)
+    return total_score
+
+
+if __name__ == "__main__":
+    try:
+        print(get_answer(sys.argv[1]))
+    except IndexError:
+        pass  # Don't crash if no input was passed through command line arguments.

puzzles/solutions/2022/d02/p2.py

@@ -0,0 +1,44 @@
+import sys
+
+import consts
+import p1
+
+
+def choose_shape(opponent_move: str, outcome: str) -> str:
+    """
+    :param opponent_move: move the opponent played
+    :param outcome: desired outcome of the move
+    :return: shape to play for the desired outcome to happen
+    """
+    mapping = consts.OUTCOME_TO_MAPPING[outcome]
+    for our_move in mapping:
+        if mapping[our_move] == opponent_move:
+            return our_move
+
+
+def generate_moves(
+    moves_outcomes: tuple[tuple[str, str]]
+) -> tuple[tuple[str, str], ...]:
+    """
+    :param moves_outcomes: list of (opponent move, desired outcome) pairs
+    :return: list of (opponent move, desired move for desired outcome to happen) pairs
+    """
+    return tuple(
+        (opponent_move, choose_shape(opponent_move, outcome))
+        for opponent_move, outcome in moves_outcomes
+    )
+
+
+def get_answer(input_text: str):
+    """Return the total score if everything goes exactly according to our actual strategy guide."""
+    moves_outcomes = p1.get_moves(input_text)
+    moves = generate_moves(moves_outcomes)
+    total_score = sum(p1.calculate_move_score(move) for move in moves)
+    return total_score
+
+
+if __name__ == "__main__":
+    try:
+        print(get_answer(sys.argv[1]))
+    except IndexError:
+        pass  # Don't crash if no input was passed through command line arguments.