2023: d08: ex1: add solution

2023/d08/ex1/ex1.py

@@ -0,0 +1,42 @@
+#!/usr/bin/env python
+
+import itertools
+import sys
+
+Graph = dict[str, list[str]]
+
+
+def solve(input: list[str]) -> int:
+    def parse_graph(input: list[str]) -> Graph:
+        res: Graph = {}
+
+        for line in input:
+            start, dests = line.split(" = ")
+            res[start] = dests[1:-1].split(", ")
+
+        return res
+
+    def parse(input: list[str]) -> tuple[str, Graph]:
+        return input[0], parse_graph(input[2:])
+
+    def navigate(directions: str, graph: Graph, start: str, end: str) -> int:
+        pos = start
+        i = 0
+        for dir in itertools.cycle(directions):
+            if pos == end:
+                break
+            pos = graph[pos][0 if dir == "L" else 1]
+            i += 1
+        return i
+
+    directions, graph = parse(input)
+    return navigate(directions, graph, "AAA", "ZZZ")
+
+
+def main() -> None:
+    input = sys.stdin.read().splitlines()
+    print(solve(input))
+
+
+if __name__ == "__main__":
+    main()