advent/y2019/p21.py

import sys

from intcode import compute


walk = [
    'NOT A J',
    'NOT C T',
    'AND D T',
    'OR T J',
    'WALK',
]
run = [
    'NOT A J',
    'NOT C T',
    'AND D T',
    'AND H T',
    'OR T J',
    'NOT B T',
    'AND D T',
    'OR T J',
    'RUN',
]
text = sys.stdin.read()
for code in [walk, run]:
    *chars, ans = compute(text, iter('\n'.join(code + ['']).encode()))
    if ans == 10:
        print(bytearray(chars).decode())
    else:
        print(ans)