5 lines
461 B
Python
5 lines
461 B
Python
import re;m,s={},"0"*36;c=lambda r,s:[i for j in((0,1)if r[0]=='X'else(int(r[0]),))for i in c(r[1:],s*2+j)]if len(r)else[s]
|
|
for(addr,dec),s in map(lambda k:(list(map(int,getattr(re.match(r"mem\[(.+)] = (.+)",k),'groups',lambda:(-1,0))())),(re.match(r"mask = (.+)",k)or(0,s))[1]),(x.strip()for x in open("input.txt"))):
|
|
m={**m,**dict((x,dec) for x in(c([(a if b=='0'else b)for a,b in zip(format(addr,'036b'),s)],0)if addr!=-1else[]))}
|
|
print(sum(m.values()))
|