56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
import re
|
|
import sys
|
|
from itertools import combinations_with_replacement
|
|
|
|
|
|
def main():
|
|
lines = [line.rstrip() for line in sys.stdin]
|
|
|
|
# Part 1
|
|
int_len = 36
|
|
meme = {}
|
|
mem_re = re.compile(r"mem\[(\d+)\] = (\d+)")
|
|
for line in lines:
|
|
if line.startswith("mask"):
|
|
mask_str = line.split("=")[1].strip()
|
|
elif (match := mem_re.match(line)):
|
|
addr, value = map(int, match.groups())
|
|
meme[addr] = apply_mask_p1(mask_str, value)
|
|
print("Init sum:", sum(meme.values()))
|
|
|
|
# Part 2 - what a mess
|
|
meme = {}
|
|
for line in lines:
|
|
if line.startswith("mask"):
|
|
mask_str = line.split("=")[1].strip()
|
|
ow_mask = int(mask_str.replace("X", "0"), 2)
|
|
floats = [i for i in range(int_len) if mask_str[::-1][i] == "X"]
|
|
elif (match := mem_re.match(line)):
|
|
addr, value = map(int, match.groups())
|
|
for masked_addr in apply_mask_p2(ow_mask, floats, addr):
|
|
meme[masked_addr] = value
|
|
print("Init sum p2:", sum(meme.values()))
|
|
|
|
|
|
def apply_mask_p1(mask, value):
|
|
value &= int(mask.replace("X", "1"), 2)
|
|
return value | int(mask.replace("X", "0"), 2)
|
|
|
|
|
|
def apply_mask_p2(ow_mask, floats, value):
|
|
value |= ow_mask
|
|
for bitbuffer in range(2 ** len(floats)):
|
|
new_value = value
|
|
for i, f in enumerate(floats):
|
|
bitbuffer_pos = 1 << i
|
|
bit_pos = 1 << f
|
|
if bitbuffer & bitbuffer_pos != 0:
|
|
new_value |= bit_pos
|
|
else:
|
|
new_value &= ~bit_pos
|
|
yield new_value
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|