sebse/python-aoc-2020
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Day 14 (glamorized)

Browse Source
This commit is contained in:
Sebastian Seedorf
2020-12-14 10:49:11 +01:00
parent 425d6ee740
commit 00801bcb48
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
day14/part1.py
View File

@@ -3,19 +3,19 @@ import re
lines = (x.strip() for x in open("input.txt"))
mem = {}
regex_mask = re.compile(r"mask = (.+)")
regex_mem = re.compile(r"mem\[(.*)] = (.+)")
regex_mem = re.compile(r"mem\[(.+)] = (.+)")
mask_and = 0
mask_or = -1
for line in lines:
match = regex_mask.match(line)
if match is not None:
mask = match.group(1)[::-1]
mask_and = sum(2**i for i, v in enumerate(mask) if v != '0')
mask_or = sum(2**i for i, v in enumerate(mask) if v == '1')
mask = match.group(1)
mask_and = int(mask.replace('X', '1'), 2)
mask_or = int(mask.replace('X', '0'), 2)
else:
match = regex_mem.match(line)
addr, dec = list(map(int, match.groups()))
mem[addr] = dec & mask_and | mask_or
print(sum(map(lambda x: mem[x], mem)))
print(sum(mem.values()))

4
day14/part2.py
View File

@@ -3,7 +3,7 @@ import re
lines = (x.strip() for x in open("input.txt"))
mem = {}
regex_mask = re.compile(r"mask = (.+)")
regex_mem = re.compile(r"mem\[(.*)] = (.+)")
regex_mem = re.compile(r"mem\[(.+)] = (.+)")
mask = ""
@@ -31,4 +31,4 @@ for line in lines:
for x in combination(res):
mem[x] = dec
print(sum(map(lambda x: mem[x], mem)))
print(sum(mem.values()))