2016: d02: ex1: add solution

2016/d02/ex1/ex1.py

@@ -0,0 +1,69 @@
+#!/usr/bin/env python
+
+import enum
+import sys
+from typing import NamedTuple
+
+
+class Point(NamedTuple):
+    x: int
+    y: int
+
+
+class Direction(enum.StrEnum):
+    UP = "U"
+    DOWN = "D"
+    LEFT = "L"
+    RIGHT = "R"
+
+    def apply(self, p: Point) -> Point:
+        match self:
+            case Direction.UP:
+                dx, dy = -1, 0
+            case Direction.DOWN:
+                dx, dy = 1, 0
+            case Direction.LEFT:
+                dx, dy = 0, -1
+            case Direction.RIGHT:
+                dx, dy = 0, 1
+        return Point(p.x + dx, p.y + dy)
+
+
+def solve(input: str) -> int:
+    def parse(input: str) -> list[list[Direction]]:
+        return [[Direction(c) for c in line] for line in input.splitlines()]
+
+    def enter_code(directions: list[list[Direction]]) -> int:
+        KEYPAD = {
+            Point(-1, -1): 1,
+            Point(-1, 0): 2,
+            Point(-1, 1): 3,
+            Point(0, -1): 4,
+            Point(0, 0): 5,
+            Point(0, 1): 6,
+            Point(1, -1): 7,
+            Point(1, 0): 8,
+            Point(1, 1): 9,
+        }
+        res = 0
+        pos = Point(0, 0)
+        for line in directions:
+            for d in line:
+                new_pos = d.apply(pos)
+                if new_pos not in KEYPAD:
+                    continue
+                pos = new_pos
+            res = res * 10 + KEYPAD[pos]
+        return res
+
+    directions = parse(input)
+    return enter_code(directions)
+
+
+def main() -> None:
+    input = sys.stdin.read()
+    print(solve(input))
+
+
+if __name__ == "__main__":
+    main()