day 7 part 1

day7/day7.py

@@ -3,27 +3,24 @@ from collections import Counter
 def calculate_rank(hand):
     card_ranks = {'2': 1, '3': 2, '4': 3, '5': 4, '6': 5, '7': 6, '8': 7, '9': 8, 'T': 9, 'J': 10, 'Q': 11, 'K': 12, 'A': 13} 
     # substitute cards with their ranks
-    hand = sorted(card_ranks.get(c, None) or c for c in hand)[::-1]
+    hand = [card_ranks[c] for c in hand]
     cnt = Counter(hand)
-    cnt_values = sorted(cnt.values())
-    rank = {
-        "5": 10,
-        "14": 9,
-        "23": 8,
-        "113": 7,
-        "122": 6,
-        "1112": 5,
-        "11111": 4,
-    }["".join(map(str, cnt_values))]
-    # return rank and sorted hand as a tiebreaker
+    rank = 0
+    match sorted(cnt.values()):
+        case [5]: rank = 7
+        case [1, 4]: rank = 6
+        case [2, 3]: rank = 5
+        case [1, 1, 3]: rank = 4
+        case [1, 2, 2]: rank = 3
+        case [1, 1, 1, 2]: rank = 2
+        case [1, 1, 1, 1, 1]: rank = 1
+    
     return (rank, hand)
 
 
 def part1(inp):
     total = 0
     hands = [l.strip().split() for l in inp.strip().split("\n")]
-    print(hands[-10: -1])
-    breakpoint()
     hands = sorted(hands, key=lambda hb: calculate_rank(hb[0]))
     for rank, hand in enumerate(hands):
         hand, bid = hand
@@ -39,11 +36,12 @@ if __name__ == "__main__":
         KTJJT 220
         QQQJA 483
     """
-    assert part1(sample_input) == 6440
+    res = part1(sample_input)
+    print(f"part 1 example: {res}")
+    assert res == 6440
     
     import sys
     if len(sys.argv) == 2:
         with open(sys.argv[1]) as f:
             res = part1(f.read())
             print(f"Part 1, res={res}")
-