Implement day 1

2024/src/aoc/days/day1.py

@@ -1,13 +1,28 @@
+from collections import defaultdict
+from io import StringIO
 from typing import Any
 
-from . import SeparateRunner
+import numpy
+
+from . import CombinedRunner
 
 
-class DayRunner(SeparateRunner):
+class DayRunner(CombinedRunner):
     @classmethod
-    def part1(cls, _data: str) -> Any:
-        return "Hello"
+    def run_both(cls, data: str) -> tuple[Any, Any]:
+        data = StringIO(data)
+        nums = numpy.loadtxt(data, dtype=numpy.int32)
 
-    @classmethod
-    def part2(cls, _data: str) -> Any:
-        return "world!"
+        left = nums[..., 0]
+        right = nums[..., 1]
+
+        left.sort()
+        right.sort()
+
+        diff = numpy.abs(left - right).sum()
+
+        counts = defaultdict(int)
+        for val in right:
+            counts[val] += 1
+
+        return diff, sum(counts[v] * v for v in left)