2024 Dec 01st

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)