diff options
Diffstat (limited to '2024/01/python.py')
-rwxr-xr-x | 2024/01/python.py | 14 |
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) |