simplify python file structure

1 files changed, 30 insertions, 0 deletions

diff --git a/2024/05/python.py b/2024/05/python.py
new file mode 100755
index 0000000..9e81978
--- /dev/null
+++ b/2024/05/python.py
@@ -0,0 +1,30 @@
+#!/usr/bin/env python3
+
+from fileinput import input
+from functools import cmp_to_key
+from itertools import takewhile
+
+inp = map(str.strip, input())
+ordering = set(takewhile(bool, inp))
+us = list(inp)
+
+
+def cmp(x, y):
+    return (f"{y}|{x}" in ordering) * 2 - 1  # hehe
+
+
+silver = 0
+gold = 0
+
+for u in us:
+    pre = u.split(",")
+    post = list(sorted(pre, key=cmp_to_key(cmp)))
+
+    mid = int(post[len(post) // 2])
+    if pre == post:
+        silver += mid
+    else:
+        gold += mid
+
+print("silver:", silver)
+print("gold:", gold)