aoc2023_21_part2_try1.py

R,E,N={0},enumerate,26501365
for M,l in E(open("input.txt")):
 for x,c in E(l.strip()):
  if'.'>c:R|={(x,M)}
  if'A'<c:q={(x,M)}
M+=1;t,h=0,[]
while len(h)<3:q={(x+u,y+v)for x,y in q for u,v in((1,0),(-1,0),(0,1),(0,-1))if(((x+u)%M,(y+v)%M)in R)-1};t+=1;h+=[len(q)]*(t%M==N%M)
a,b,c=h;N//=M
print(a+N*(4*b-3*a-c+N*(c-2*b+a))//2)