| Total Complexity | 8 | 
| Total Lines | 83 | 
| Duplicated Lines | 0 % | 
| Changes | 0 | ||
| 1 | from __future__ import annotations | ||
| 2 | |||
| 3 | import pathlib | ||
| 4 | from enum import IntEnum | ||
| 5 | from typing import Tuple, List, Dict | ||
| 6 | |||
| 7 | WIN_SCORE = 6 | ||
| 8 | DRAW_SCORE = 3 | ||
| 9 | |||
| 10 | |||
| 11 | class PlayMove(IntEnum): | ||
| 12 | ROCK = 1 | ||
| 13 | PAPER = 2 | ||
| 14 | SCISSORS = 3 | ||
| 15 | |||
| 16 | @property | ||
| 17 | def win(self) -> PlayMove: | ||
| 18 | return PlayMove((self + 1) % 4 or 1) | ||
| 19 | |||
| 20 | @property | ||
| 21 | def lose(self) -> PlayMove: | ||
| 22 | return PlayMove((self - 1) or 3) | ||
| 23 | |||
| 24 | |||
| 25 | class Target(IntEnum): | ||
| 26 | LOSE = 1 | ||
| 27 | DRAW = 2 | ||
| 28 | WIN = 3 | ||
| 29 | |||
| 30 | def score_from_enemy(self, enemy: PlayMove) -> int: | ||
| 31 | if self == Target.LOSE: | ||
| 32 | return enemy.lose | ||
| 33 | |||
| 34 | return ( | ||
| 35 | enemy.win + WIN_SCORE | ||
| 36 | if self == Target.WIN | ||
| 37 | else enemy + DRAW_SCORE | ||
| 38 | ) | ||
| 39 | |||
| 40 | |||
| 41 | TRANSLATION_MAP: Dict[str, PlayMove] = { | ||
| 42 | 'A': PlayMove.ROCK, | ||
| 43 | 'B': PlayMove.PAPER, | ||
| 44 | 'C': PlayMove.SCISSORS, | ||
| 45 | |||
| 46 | 'X': PlayMove.ROCK, | ||
| 47 | 'Y': PlayMove.PAPER, | ||
| 48 | 'Z': PlayMove.SCISSORS, | ||
| 49 | } | ||
| 50 | |||
| 51 | |||
| 52 | def get_part_one_score(moves: List[Tuple[PlayMove, ...]]) -> int: | ||
| 53 | return sum( | ||
| 54 | ( | ||
| 55 | my_move | ||
| 56 | + (enemy_move.win == my_move) * WIN_SCORE | ||
| 57 | + (my_move == enemy_move) * DRAW_SCORE | ||
| 58 | ) for enemy_move, my_move in moves | ||
| 59 | ) | ||
| 60 | |||
| 61 | |||
| 62 | def get_part_two_score(moves: List[Tuple[PlayMove, ...]]) -> int: | ||
| 63 | return sum( | ||
| 64 | Target(target).score_from_enemy(enemy_move) | ||
| 65 | for enemy_move, target in moves | ||
| 66 | ) | ||
| 67 | |||
| 68 | |||
| 69 | def main(): | ||
| 70 |     content = pathlib.Path('./input.txt').read_text() | ||
| 71 | |||
| 72 | moves: List[Tuple[PlayMove, ...]] = [ | ||
| 73 | tuple(map(TRANSLATION_MAP.get, line.split())) | ||
| 74 | for line in content.splitlines() | ||
| 75 | ] | ||
| 76 | |||
| 77 |     print("Part 1:", get_part_one_score(moves)) | ||
| 78 |     print("Part 2:", get_part_two_score(moves)) | ||
| 79 | |||
| 80 | |||
| 81 | if __name__ == '__main__': | ||
| 82 | main() | ||
| 83 |