From af6a172e812b32c6b314af4fd01cd0845f84c59d Mon Sep 17 00:00:00 2001
From: npc-strider <npc.strider@gmail.com>
Date: Mon, 14 Dec 2020 14:01:41 +0800
Subject: [PATCH] day 14

---
 14/a.py | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 14/a.py

diff --git a/14/a.py b/14/a.py
new file mode 100644
index 0000000..1ca9ac3
--- /dev/null
+++ b/14/a.py
@@ -0,0 +1,52 @@
+
+from itertools import product
+
+d = [ x.replace('[', ' = ').replace(']', '').split(' = ') for x in open('input').read().splitlines() ]
+
+def sum(ram, b):
+    c = 0
+    for x in ram:
+        # print(ram[x])
+        c += int(ram[x], b)
+    return c
+    # print(sum({int(ram[s], 2) for s in ram}))
+
+
+def a(d):
+    ram = {}
+    mask = []
+    for instruction in d:
+        cmd = instruction[0]
+        # print(instruction)
+        if cmd == 'mask':
+            mask = list(instruction[1])
+        elif cmd == 'mem':
+            # print(int(instruction[1]))
+            b = list(bin(int(instruction[2]))[2:].zfill(36))
+            for x in range(len(mask)):
+                if mask[x] != 'X':
+                    b[x] = mask[x]
+            ram[int(instruction[1])] = ''.join(b)
+    return sum(ram, 2)
+
+def b(d):
+    ram = {}
+    mask = []
+    for instruction in d:
+        cmd = instruction[0]
+        # print(instruction)
+        if cmd == 'mask':
+            mask = list(instruction[1])
+        elif cmd == 'mem':
+            b = list(bin(int(instruction[1]))[2:].zfill(36))
+            for x in range(len(mask)):
+                if mask[x] != '0':
+                    b[x] = mask[x]
+            b = ''.join(b)
+            b_ft = b.replace('X', '{}') 
+            for i in product('01', repeat=b.count('X')):
+                addr = int(b_ft.format(*i),2)
+                ram[addr] = instruction[2]
+    return sum(ram, 10)
+
+print(a(d), b(d))
\ No newline at end of file