2023: d08: ex1: add solution

This commit is contained in:
Bruno BELANYI 2023-12-08 08:23:05 +00:00
parent dcce8c1eec
commit 34e16218b8

42
2023/d08/ex1/ex1.py Executable file
View file

@ -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()