aoc2019_19_part1.py

O=map(int,open("input.txt").read().split(','))
def S(x,y):
 M=dict(enumerate(O));p=s=0
 while 1:
  o=M[p];l=map(int,str(o)[-3::-1]+"000");o%=100;n=int("0331122331"[o]);i,j,k=[M.get(p+u,0)+s*(m>1)for u,m in zip((1,2,3),l)];a,b=[(M.get(u,0)if m-1else u)for u,m in zip((i,j),l)];p+=n+1
  if o<2:M[k]=a+b
  elif o<3:M[k]=a*b
  elif o<4:M[i]=x;x=y
  elif o<5:return a
  elif o>8:s+=a
  elif o>7:M[k]=a==b
  elif o>6:M[k]=a<b
  elif(o>5)^(a!=0):p=b
R=range(50)
print sum(S(x,y)for x in R for y in R)