2024: d03: ex1: add solution

2024/d03/ex1/ex1.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python
+
+import dataclasses
+import re
+import sys
+
+
+@dataclasses.dataclass
+class MulInstruction:
+    lhs: int
+    rhs: int
+
+    def calc(self) -> int:
+        return self.lhs * self.rhs
+
+
+def solve(input: str) -> int:
+    def parse(input: str) -> list[MulInstruction]:
+        res: list[MulInstruction] = []
+
+        MUL_REGEX = re.compile(r"mul\((\d+),(\d+)\)")
+        for match in MUL_REGEX.finditer(input):
+            res.append(MulInstruction(int(match.group(1)), int(match.group(2))))
+
+        return res
+
+    return sum(inst.calc() for inst in parse(input))
+
+
+def main() -> None:
+    input = sys.stdin.read()
+    print(solve(input))
+
+
+if __name__ == "__main__":
+    main()