30 lines
896 B
Python
30 lines
896 B
Python
#!/usr/bin/env python3
|
|
from collections import Counter
|
|
|
|
lines = (x.strip() for x in open("input.txt"))
|
|
|
|
hands = []
|
|
|
|
for line in lines:
|
|
hand, bid = line.split(" ")
|
|
hand = tuple("J23456789TQKA".index(char) for char in hand)
|
|
count = list(Counter(h for h in hand if h != 0).values())
|
|
num_j = sum(1 for h in hand if h == 0)
|
|
if len(count) <= 1:
|
|
hands.append((6, hand, int(bid)))
|
|
elif (4-num_j) in count:
|
|
hands.append((5, hand, int(bid)))
|
|
elif len(count) == 2 and 2 in count:
|
|
hands.append((4, hand, int(bid)))
|
|
elif (3-num_j) in count:
|
|
hands.append((3, hand, int(bid)))
|
|
elif len(count) == 3:
|
|
hands.append((2, hand, int(bid)))
|
|
elif len(count) == 4:
|
|
hands.append((1, hand, int(bid)))
|
|
else:
|
|
hands.append((0, hand, int(bid)))
|
|
|
|
hands.sort()
|
|
print(sum(bid * rank for rank, (_, _, bid) in enumerate(hands, 1)))
|