From 73ee756062cb97b396bf6aaec05423751c9b04d3 Mon Sep 17 00:00:00 2001
From: Bruno BELANYI <bruno@belanyi.fr>
Date: Thu, 7 Dec 2023 08:56:39 +0000
Subject: [PATCH] 2023: d07: ex1: add solution

---
 2023/d07/ex1/ex1.py | 56 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100755 2023/d07/ex1/ex1.py

diff --git a/2023/d07/ex1/ex1.py b/2023/d07/ex1/ex1.py
new file mode 100755
index 0000000..8c7a00e
--- /dev/null
+++ b/2023/d07/ex1/ex1.py
@@ -0,0 +1,56 @@
+#!/usr/bin/env python
+
+import functools
+import sys
+from collections import Counter
+
+Play = tuple[str, int]
+
+ORDER = "23456789TJQKA"
+
+
+def solve(input: list[str]) -> int:
+    def parse(input: list[str]) -> list[Play]:
+        return [(hand, int(bid)) for hand, bid in map(str.split, input)]
+
+    # Stronger hands compare higher than weaker hands
+    def cmp_hand(lhs: str, rhs: str) -> int:
+        def to_type(hand: str) -> list[int]:
+            return [count for _, count in Counter(hand).most_common()]
+
+        def cmp_type(lhs: list[int], rhs: list[int]) -> int:
+            for left, right in zip(lhs, rhs):
+                if (cmp := left - right) != 0:
+                    return cmp
+            return 0
+
+        def cmp_card(lhs: str, rhs: str) -> int:
+            assert lhs in ORDER and rhs in ORDER  # Sanity check
+            assert len(lhs) == 1 and len(rhs) == 1  # Sanity check
+            return ORDER.find(lhs) - ORDER.find(rhs)
+
+        if (cmp := cmp_type(to_type(lhs), to_type(rhs))) != 0:
+            return cmp
+
+        for left, right in zip(lhs, rhs):
+            if (cmp := cmp_card(left, right)) != 0:
+                return cmp
+
+        return 0
+
+    def cmp_play(lhs: Play, rhs: Play) -> int:
+        return cmp_hand(lhs[0], rhs[0])
+
+    plays = parse(input)
+    plays = sorted(plays, key=functools.cmp_to_key(cmp_play))
+
+    return sum(rank * bid for rank, (_, bid) in enumerate(plays, start=1))
+
+
+def main() -> None:
+    input = sys.stdin.read().splitlines()
+    print(solve(input))
+
+
+if __name__ == "__main__":
+    main()