Day 07

day07/part1.py

@@ -0,0 +1,28 @@
+#!/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("23456789TJQKA".index(char) for char in hand)
+    count = Counter(hand).values()
+    if 5 in count:
+        hands.append((6, hand, int(bid)))
+    elif 4 in count:
+        hands.append((5, hand, int(bid)))
+    elif 3 in count and 2 in count:
+        hands.append((4, hand, int(bid)))
+    elif 3 in count:
+        hands.append((3, hand, int(bid)))
+    elif list(count).count(2) == 2:
+        hands.append((2, hand, int(bid)))
+    elif 2 in count:
+        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)))

day07/part2.py

@@ -0,0 +1,31 @@
+#!/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 list(count).count(2) == 2:
+        hands.append((2, hand, int(bid)))
+    elif (2-num_j) in count:
+        hands.append((1, hand, int(bid)))
+    else:
+        hands.append((0, hand, int(bid)))
+
+hands.sort()
+for h in hands:
+    print(h)
+print(sum(bid * rank for rank, (_, _, bid) in enumerate(hands, 1)))