43 lines
1.1 KiB
Ruby
43 lines
1.1 KiB
Ruby
class PokerHand
|
|
include Comparable
|
|
|
|
attr_reader :hand, :bet, :type
|
|
|
|
def initialize(hand, bet)
|
|
@hand = hand.gsub('A', 'Z').gsub('K', 'Y').gsub('Q', 'X').gsub('J', '0').gsub('T', 'V')
|
|
@bet = bet
|
|
@type = get_hand_type hand.gsub('J', '')
|
|
end
|
|
|
|
def get_hand_type hand
|
|
num_unique_cards = hand.chars.uniq.length
|
|
num_jokers = 5 - hand.length
|
|
unique_map = hand.chars.group_by(&:itself).map { |_char, instances| instances.length }
|
|
return 1 if num_unique_cards == 1 || num_jokers == 5
|
|
return 2 if (num_unique_cards == 2) && unique_map.include?(4 - num_jokers)
|
|
return 3 if num_unique_cards == 2
|
|
return 4 if (num_unique_cards == 3) && unique_map.include?(3 - num_jokers)
|
|
return 5 if num_unique_cards == 3
|
|
return 6 if num_unique_cards == 4
|
|
|
|
7
|
|
end
|
|
|
|
def <=>(other)
|
|
if @type == other.type
|
|
@hand <=> other.hand
|
|
else
|
|
other.type <=> @type
|
|
end
|
|
end
|
|
end
|
|
|
|
hands = File.foreach('./2023/day7/data').map { |line| PokerHand.new(line.split(' ')[0], line.split(' ')[1].to_i) }
|
|
|
|
hands.sort!
|
|
ret = 0
|
|
hands.each_with_index do |hand, i|
|
|
ret += hand.bet * (i + 1)
|
|
end
|
|
puts ret
|