73 lines
1.5 KiB
Python
73 lines
1.5 KiB
Python
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
|