day 24: pythonHEADmaster

1 files changed, 28 insertions, 0 deletions

diff --git a/24.py b/24.py
new file mode 100644
index 0000000..fee2116
--- /dev/null
+++ b/24.py
@@ -0,0 +1,28 @@
+from utils import open_day
+
+digits = dict()
+stack = list()
+
+with open_day(24) as f:
+    dig = 0
+    for i, line in enumerate(f):
+        _, *operands = line.rstrip().split(' ')
+        if i % 18 == 4: push = operands[1] == '1'
+        if i % 18 == 5: sub = int(operands[1])
+        if i % 18 == 15:
+            add = int(operands[1])
+            if push:
+                stack.append((dig, add))
+            else:
+                sibling, add = stack.pop()
+                diff = add + sub
+                if diff < 0:
+                    digits[sibling] = (-diff + 1, 9)
+                    digits[dig] = (1, 9 + diff)
+                else:
+                    digits[sibling] = (1, 9 - diff)
+                    digits[dig] = (1 + diff, 9)
+            dig += 1
+
+print(''.join(str(digits[d][1]) for d in sorted(digits.keys())))
+print(''.join(str(digits[d][0]) for d in sorted(digits.keys())))