Jake Pullen
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
with open(r'advent_of_code\2023\day_7\input.txt', 'r') as file:
|
|
input = file.read()
|
|
|
|
#print(input)
|
|
|
|
test_input = '''32T3K 765
|
|
T55J5 684
|
|
KK677 28
|
|
KTJJT 220
|
|
QQQJA 483'''
|
|
|
|
#input = test_input
|
|
|
|
# Split the input into lines
|
|
lines = input.split('\n')
|
|
|
|
# Process each line
|
|
processed_lines = []
|
|
for line in lines:
|
|
# Strip leading/trailing whitespace and split the line into parts
|
|
parts = line.strip().split()
|
|
# Convert the second part to an integer and add the processed line to the list
|
|
processed_lines.append((parts[0], int(parts[1])))
|
|
|
|
# Convert the list of processed lines to a tuple
|
|
hands = tuple(processed_lines)
|
|
|
|
cards = ['2', '3', '4', '5', '6', '7', '8', '9', '10', 'T', 'J', 'Q', 'K', 'A']
|
|
|
|
def score(hand):
|
|
# Create a set from the hand to remove duplicates
|
|
unique_chars = set(hand)
|
|
|
|
# Count the occurrences of each unique character in the hand
|
|
char_counts = [hand.count(x) for x in unique_chars]
|
|
|
|
# Sort the counts in descending order
|
|
sorted_counts = sorted(char_counts, reverse=True)
|
|
|
|
# If there are no counts (i.e., the hand was empty), use [5] as the default
|
|
sorted_counts = [5] if sorted_counts == [] else sorted_counts
|
|
|
|
# Convert the counts to strings, join them into a single string, and pad it with zeros on the right
|
|
count_str = ''.join(str(c) for c in sorted_counts).ljust(5, '0')
|
|
|
|
# For each character in the hand, add its index in the 'cards' list (plus 1) to the string
|
|
for x in hand:
|
|
count_str += str(cards.index(x) + 1).zfill(2)
|
|
|
|
# Convert the final string to an integer and return it
|
|
return int(count_str)
|
|
|
|
|
|
def total_winnings(hands):
|
|
# Sort the hands based on their score
|
|
sorted_hands = sorted(hands, key=lambda x: score(x[0]))
|
|
|
|
# Calculate the winnings for each hand
|
|
winnings = []
|
|
for i, hand in enumerate(sorted_hands):
|
|
# The winnings for a hand is its rank (i + 1) times its bet (hand[1])
|
|
hand_winnings = (i + 1) * hand[1]
|
|
winnings.append(hand_winnings)
|
|
|
|
# Return the total winnings
|
|
return sum(winnings)
|
|
|
|
|
|
print(total_winnings(hands)) |