124 lines
3.8 KiB
Python
124 lines
3.8 KiB
Python
from __future__ import annotations
|
|
|
|
from enum import Enum
|
|
from collections import defaultdict
|
|
from functools import reduce
|
|
|
|
from .aoc import AocParseLinesSameParser, 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 Hand2:
|
|
def __init__(self, string: str) -> None:
|
|
translation = "J23456789TQKA"
|
|
self.string = [translation.index(x) for x in string]
|
|
self.cards_dict = 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 !="J"]))))
|
|
joker_amount = self.cards_dict["J"]
|
|
|
|
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 Hand:
|
|
def __init__(self, string: str) -> None:
|
|
translation = "23456789TJQKA"
|
|
self.string = [translation.index(x) for x in string]
|
|
self.cards_dict = defaultdict(lambda : 0)
|
|
for letter in string:
|
|
self.cards_dict[letter] += 1
|
|
|
|
value_amount = list(reversed(sorted(list(self.cards_dict.values()))))
|
|
|
|
match value_amount[0]:
|
|
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(AocParseLinesSameParser[tuple[Hand, int]], Aoc2[list[tuple[Hand, int]], list[tuple[Hand, int]]]):
|
|
class Day07(Aoc2[list[tuple[Hand, int]], list[tuple[Hand2, int]]]):
|
|
def parseinput1(self, inpt: str) -> list[tuple[Hand, int]]:
|
|
lines = inpt.splitlines()
|
|
return [(Hand(line.split(" ")[0]), int(line.split(" ")[1])) for line in lines]
|
|
|
|
def parseinput2(self, inpt: str) -> list[tuple[Hand2, int]]:
|
|
lines = inpt.splitlines()
|
|
return [(Hand2(line.split(" ")[0]), 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[Hand2, 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()
|
|
|