Solve 2016:12 p1-2 "Leonardo's Monorail"

2016-python2/output/day_12.py

@@ -0,0 +1,73 @@
+from output import answer  # , matrix, D, DD, ADJ, ints, mhd, mdbg, vdbg
+
+n = 12
+title = "Leonardo's Monorail"
+
+
+@answer(1, "Value of registry a will be {} on exit")
+def part_1(presolved):
+    return presolved[0]
+
+
+@answer(
+    2, "If register c is initialized with 1, value of registry a will be {} on exit"
+)
+def part_2(presolved):
+    return presolved[1]
+
+
+def solve(data):
+    p = data.splitlines()
+    pl = len(p)
+
+    def run(r):
+        i = 0
+        while i < pl:
+            o, *a = p[i].split()
+            match o:
+                case "cpy":
+                    kv, t = a
+                    r[t] = int(kv) if kv.isdigit() else r[kv]
+                    i += 1
+                case "jnz":
+                    kv, d = a
+                    c = int(kv) if kv.isdigit() else r[kv]
+                    i += 1 if c == 0 else int(d)
+                case "inc":
+                    r[a[0]] += 1
+                    i += 1
+                case "dec":
+                    r[a[0]] -= 1
+                    i += 1
+        return r["a"]
+
+    p1 = run(
+        {
+            "a": 0,
+            "b": 0,
+            "c": 0,
+            "d": 0,
+        }
+    )
+    p2 = run(
+        {
+            "a": 0,
+            "b": 0,
+            "c": 1,
+            "d": 0,
+        }
+    )
+    return p1, p2
+
+
+if __name__ == "__main__":
+    with open("./input/12.txt", "r") as f:
+        inp = f.read().strip()
+
+    inp = solve(inp)
+
+    a = part_1(inp)
+    b = part_2(inp)
+
+    assert a == 318009
+    assert b == 9227663