70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
#!/usr/bin/env python
|
|
# 2023 - Advent Of Code 7
|
|
|
|
# file = 'input_example.txt'
|
|
file = 'input.txt'
|
|
|
|
|
|
class Hand:
|
|
def __init__(self, cards, bet):
|
|
self.cards = cards
|
|
self.bet = bet
|
|
self.dict_cards = {}
|
|
|
|
def __repr__(self):
|
|
return repr(f'cards: {self.cards}')
|
|
|
|
def sort_cards(self):
|
|
self.dict_cards = sorted(self.dict_cards.items(), key=lambda x: x[1], reverse=True)
|
|
|
|
|
|
cards_order = ['2', '3', '4', '5', '6', '7', '8', '9', 'T', 'J', 'Q', 'K', 'A']
|
|
|
|
# pylint: disable=consider-using-with
|
|
input_lines = [line.strip('\n') for line in open(file, encoding="utf-8")]
|
|
|
|
hands = [Hand(line.split()[0], line.split()[1]) for line in input_lines]
|
|
print(hands)
|
|
|
|
for h in hands:
|
|
hand_dict = {k: 0 for k in cards_order}
|
|
for card in h.cards:
|
|
hand_dict[card] += 1
|
|
h.dict_cards = {k: v for k, v in hand_dict.items() if v != 0}
|
|
|
|
for h in hands:
|
|
h.sort_cards()
|
|
|
|
hands_5 = [h for h in hands if h.dict_cards[0][1] == 5]
|
|
# for h in hands_5:
|
|
# print(h)
|
|
|
|
sorted_5 = sorted(hands_5, key=lambda x: [cards_order.index(c) for c in x.cards])
|
|
|
|
hands_4 = [h for h in hands if h.dict_cards[0][1] == 4]
|
|
sorted_4 = sorted(hands_4, key=lambda x: [cards_order.index(c) for c in x.cards])
|
|
|
|
hands_full = [h for h in hands if h.dict_cards[0][1] == 3 and h.dict_cards[1][1] == 2]
|
|
sorted_full = sorted(hands_full, key=lambda x: [cards_order.index(c) for c in x.cards])
|
|
|
|
hands_3 = [h for h in hands if h.dict_cards[0][1] == 3 and h.dict_cards[1][1] != 2]
|
|
sorted_3 = sorted(hands_3, key=lambda x: [cards_order.index(c) for c in x.cards])
|
|
|
|
hands_twopairs = [h for h in hands if h.dict_cards[0][1] == 2 and h.dict_cards[1][1] == 2]
|
|
sorted_twopairs = sorted(hands_twopairs, key=lambda x: [cards_order.index(c) for c in x.cards])
|
|
|
|
hands_onepair = [h for h in hands if h.dict_cards[0][1] == 2 and h.dict_cards[1][1] != 2]
|
|
sorted_onepair = sorted(hands_onepair, key=lambda x: [cards_order.index(c) for c in x.cards])
|
|
|
|
hands_high = [h for h in hands if h.dict_cards[0][1] == 1]
|
|
sorted_high = sorted(hands_high, key=lambda x: [cards_order.index(c) for c in x.cards])
|
|
|
|
sorted_hands = sorted_high + sorted_onepair + sorted_twopairs + sorted_3 + sorted_full + sorted_4 + sorted_5
|
|
|
|
acc = 0
|
|
for rank, h in enumerate(sorted_hands):
|
|
# print(f'{rank+1} {h.bet}')
|
|
print((rank + 1) * int(h.bet))
|
|
acc += (rank+1) * int(h.bet)
|
|
|
|
print(acc)
|