summaryrefslogtreecommitdiff
path: root/2024/01/python.py
diff options
context:
space:
mode:
Diffstat (limited to '2024/01/python.py')
-rwxr-xr-x2024/01/python.py14
1 files changed, 14 insertions, 0 deletions
diff --git a/2024/01/python.py b/2024/01/python.py
new file mode 100755
index 0000000..425f6c7
--- /dev/null
+++ b/2024/01/python.py
@@ -0,0 +1,14 @@
+#!/usr/bin/env python3
+
+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)