From 34e16218b83257d673b7f7a2737d9a63dbc83086 Mon Sep 17 00:00:00 2001 From: Bruno BELANYI Date: Fri, 8 Dec 2023 08:23:05 +0000 Subject: [PATCH] 2023: d08: ex1: add solution --- 2023/d08/ex1/ex1.py | 42 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100755 2023/d08/ex1/ex1.py diff --git a/2023/d08/ex1/ex1.py b/2023/d08/ex1/ex1.py new file mode 100755 index 0000000..9765985 --- /dev/null +++ b/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()