From 564643c00129436b37f97a3b5f76e2ed5f582c6f Mon Sep 17 00:00:00 2001 From: Julia Lange Date: Sun, 1 Dec 2024 00:02:18 -0800 Subject: [PATCH] 2024 Dec 01st --- Python/2024/01/main.py | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 Python/2024/01/main.py diff --git a/Python/2024/01/main.py b/Python/2024/01/main.py new file mode 100644 index 0000000..8970248 --- /dev/null +++ b/Python/2024/01/main.py @@ -0,0 +1,29 @@ +from functools import cmp_to_key +input = open("input", 'r') + +data1 = [] +data2 = [] + +for i, line in enumerate(input): + a, b = line.split(" ") + data1.append((int(a.strip()), i)) + data2.append((int(b.strip()), i)) + +data1 = sorted(data1) +data2 = sorted(data2) + +sm = 0 + +for d1,d2 in zip(data1,data2): + sm += abs(d1[0] - d2[0]) + +print(sm) + +sm2 = 0 + +data2 = [d2 for d2, _ in data2] + +for d1 in data1: + sm2 += d1[0] * data2.count(d1[0]) + +print(sm2)