99 lines
2.8 KiB
Python
99 lines
2.8 KiB
Python
from __future__ import annotations
|
|
|
|
from collections import defaultdict
|
|
from enum import Enum
|
|
from typing import Optional
|
|
|
|
from .aoc import Aoc2
|
|
|
|
|
|
class Type(Enum):
|
|
HIGH_CARD = 0
|
|
ONE_PAIR = 1
|
|
TWO_PAIR = 2
|
|
THREE_KIND = 3
|
|
FULL_HOUSE = 4
|
|
FOUR_KIND = 5
|
|
FIVE_KIND = 6
|
|
|
|
def __lt__(self, other: Type) -> bool:
|
|
return self.value < other.value
|
|
|
|
|
|
class Hand:
|
|
def __init__(
|
|
self, string: str, translation: str, joker: Optional[str] = None
|
|
) -> None:
|
|
self.string = [translation.index(x) for x in string]
|
|
self.cards_dict: dict[str, int] = defaultdict(lambda: 0)
|
|
for letter in string:
|
|
self.cards_dict[letter] += 1
|
|
|
|
value_amount = list(
|
|
reversed(
|
|
sorted(list([v for k, v in self.cards_dict.items() if k != joker]))
|
|
)
|
|
)
|
|
joker_amount = 0
|
|
if joker is not None:
|
|
joker_amount = self.cards_dict[joker]
|
|
|
|
if joker_amount == 5:
|
|
self.type = Type.FIVE_KIND
|
|
return
|
|
|
|
match value_amount[0] + joker_amount:
|
|
case 5:
|
|
self.type = Type.FIVE_KIND
|
|
case 4:
|
|
self.type = Type.FOUR_KIND
|
|
case 3:
|
|
if value_amount[1] == 2:
|
|
self.type = Type.FULL_HOUSE
|
|
else:
|
|
self.type = Type.THREE_KIND
|
|
case 2:
|
|
if value_amount[1] == 2:
|
|
self.type = Type.TWO_PAIR
|
|
else:
|
|
self.type = Type.ONE_PAIR
|
|
case _:
|
|
self.type = Type.HIGH_CARD
|
|
|
|
def __lt__(self, other: Hand) -> bool:
|
|
"""self < other"""
|
|
return (self.type, self.string) < (other.type, other.string)
|
|
|
|
def __repr__(self) -> str:
|
|
return f"Hand({self.type}, {self.string})"
|
|
|
|
|
|
class Day07(Aoc2[list[tuple[Hand, int]], list[tuple[Hand, int]]]):
|
|
def parseinput1(self, inpt: str) -> list[tuple[Hand, int]]:
|
|
lines = inpt.splitlines()
|
|
return [
|
|
(Hand(line.split(" ")[0], "23456789TJQKA"), int(line.split(" ")[1]))
|
|
for line in lines
|
|
]
|
|
|
|
def parseinput2(self, inpt: str) -> list[tuple[Hand, int]]:
|
|
lines = inpt.splitlines()
|
|
return [
|
|
(Hand(line.split(" ")[0], "J23456789TQKA", "J"), int(line.split(" ")[1]))
|
|
for line in lines
|
|
]
|
|
|
|
def part1(self, inpt: list[tuple[Hand, int]]) -> int:
|
|
return sum(((rnk + 1) * bid) for rnk, (_, bid) in list(enumerate(sorted(inpt))))
|
|
|
|
def part2(self, inpt: list[tuple[Hand, int]]) -> int:
|
|
return sum(((rnk + 1) * bid) for rnk, (_, bid) in list(enumerate(sorted(inpt))))
|
|
|
|
|
|
def main() -> None:
|
|
day07 = Day07(2023, 7)
|
|
day07.run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|