path: root/prog/aoc/23/8/1.py

#!/usr/bin/python3
inst = input()
input()
graph = {}
try:
	while True:
		a = input().split(" = ")
		graph[a[0]] = a[1].replace("(", "").replace(")", "").split(", ")
except EOFError:
	pass
cur = "AAA"
runs = 0
while True:
	for d in inst:
		if cur == "ZZZ":
			break
		if d == "L":
			cur = graph[cur][0]
		else:
			cur = graph[cur][1]
		runs += 1
	else:
		continue
	break
print(runs)
r = []
for s in [x for x in graph if x[2] == "A"]:
	runs = 0
	while True:
		for d in inst:
			if s[2] == "Z":
				break
			if d == "L":
				s = graph[s][0]
			else:
				s = graph[s][1]
			runs += 1
		else:
			continue
		break
	r.append(runs)
from math import lcm
print(lcm(*r))