day14.py

with open("day14_input.txt") as f:
    data = f.read().split('\n')


# Part 1
memory = [0] * 100000
for line in data:
    if line[0:4] == "mask":
        mask = line[7:]
        mask0 = int(''.join(map(lambda c: '0' if c == 'X' else c, mask)), 2)
        mask1 = int(''.join(map(lambda c: '1' if c == 'X' else c, mask)), 2)
    if line[0:3] == "mem":
        loc = int(line[4:line.find(']')])
        num = int(line[line.find(']') + 4:])
        memory[loc] = (num | mask0) & mask1
print(sum(memory))