2024-01 python p1,p2

1 files changed, 12 insertions, 0 deletions

diff --git a/2024/01/python/main.py b/2024/01/python/main.py
new file mode 100644
index 0000000..f920d5d
--- /dev/null
+++ b/2024/01/python/main.py
@@ -0,0 +1,12 @@
+from collections import Counter
+from fileinput import input
+
+L, R = zip(*[[int(n) for n in s.split()] for s in input()])
+
+silver = sum(abs(left - right) for left, right in zip(sorted(L), sorted(R)))
+
+L, R = Counter(L), Counter(R)
+gold = sum(n * L[n] * R[n] for n in list(L & R))
+
+print("silver:", silver)
+print("gold:", gold)