91 lines
3.9 KiB
Python
91 lines
3.9 KiB
Python
#!/usr/bin/env python
|
|
# 2023 - Advent Of Code 7 - part 2
|
|
|
|
# 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']
|
|
cards_order = ['J', '2', '3', '4', '5', '6', '7', '8', '9', 'T', '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()
|
|
# print(h.dict_cards)
|
|
|
|
hands_5 = [h for h in hands if h.dict_cards[0][1] == 5]
|
|
hands_5.extend([h for h in hands if h.dict_cards[0][1] == 4 and (('J', 1) in h.dict_cards or ('J', 4) in h.dict_cards)])
|
|
hands_5.extend([h for h in hands if h.dict_cards[0][1] == 3 and ('J', 2) in h.dict_cards])
|
|
hands_5.extend([h for h in hands if ('J', 3) in h.dict_cards and h.dict_cards[1][1] == 2])
|
|
sorted_5 = sorted(hands_5, key=lambda x: [cards_order.index(c) for c in x.cards])
|
|
filtered_hands = [f for f in hands if f not in sorted_5]
|
|
hands = filtered_hands
|
|
|
|
hands_4 = [h for h in hands if h.dict_cards[0][1] == 4]
|
|
hands_4.extend([h for h in hands if h.dict_cards[0][1] == 3 and (('J', 1) in h.dict_cards or ('J', 3) in h.dict_cards)])
|
|
hands_4.extend([h for h in hands if h.dict_cards[0][1] == 2 and h.dict_cards[1][1] == 2 and ('J', 2) in h.dict_cards])
|
|
sorted_4 = sorted(hands_4, key=lambda x: [cards_order.index(c) for c in x.cards])
|
|
filtered_hands = [f for f in hands if f not in sorted_4]
|
|
hands = filtered_hands
|
|
|
|
hands_full = [h for h in hands if h.dict_cards[0][1] == 3 and h.dict_cards[1][1] == 2]
|
|
hands_full.extend([h for h in hands if h.dict_cards[0][1] == 2 and h.dict_cards[1][1] == 2 and ('J', 1) in h.dict_cards])
|
|
sorted_full = sorted(hands_full, key=lambda x: [cards_order.index(c) for c in x.cards])
|
|
filtered_hands = [f for f in hands if f not in sorted_full]
|
|
hands = filtered_hands
|
|
|
|
hands_3 = [h for h in hands if h.dict_cards[0][1] == 3 and h.dict_cards[1][1] != 2]
|
|
hands_3.extend([h for h in hands if h.dict_cards[0][1] == 2 and ('J', 1) in h.dict_cards])
|
|
hands_3.extend([h for h in hands if h.dict_cards[0][1] == 2 and h.dict_cards[0][0] == 'J' and h.dict_cards[1][1] != 2])
|
|
sorted_3 = sorted(hands_3, key=lambda x: [cards_order.index(c) for c in x.cards])
|
|
filtered_hands = [f for f in hands if f not in sorted_3]
|
|
hands = filtered_hands
|
|
|
|
hands_twopairs = [h for h in hands if h.dict_cards[0][1] == 2 and h.dict_cards[1][1] == 2]
|
|
hands_3.extend([h for h in hands if h.dict_cards[0][1] == 2 and ('J', 1) in h.dict_cards])
|
|
sorted_twopairs = sorted(hands_twopairs, key=lambda x: [cards_order.index(c) for c in x.cards])
|
|
filtered_hands = [f for f in hands if f not in sorted_twopairs]
|
|
hands = filtered_hands
|
|
|
|
hands_onepair = [h for h in hands if h.dict_cards[0][1] == 2 and h.dict_cards[1][1] != 2]
|
|
hands_onepair.extend([h for h in hands if h.dict_cards[0][1] == 1 and ('J', 1) in h.dict_cards])
|
|
sorted_onepair = sorted(hands_onepair, key=lambda x: [cards_order.index(c) for c in x.cards])
|
|
filtered_onepair = [f for f in hands if f not in sorted_onepair]
|
|
hands = filtered_onepair
|
|
|
|
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} {h.bet}')
|
|
# print((rank + 1) * int(h.bet))
|
|
acc += (rank+1) * int(h.bet)
|
|
|
|
print(acc)
|