x = [ 3, 225, 1, 225, 6, 6, 1100, 1, 238, 225, 104, 0, 1102, 31, 68, 225, 1001, 13, 87, 224, 1001, 224, -118, 224, 4, 224, 102, 8, 223, 223, 1001, 224, 7, 224, 1, 223, 224, 223, 1, 174, 110, 224, 1001, 224, -46, 224, 4, 224, 102, 8, 223, 223, 101, 2, 224, 224, 1, 223, 224, 223, 1101, 13, 60, 224, 101, -73, 224, 224, 4, 224, 102, 8, 223, 223, 101, 6, 224, 224, 1, 224, 223, 223, 1101, 87, 72, 225, 101, 47, 84, 224, 101, -119, 224, 224, 4, 224, 1002, 223, 8, 223, 1001, 224, 6, 224, 1, 223, 224, 223, 1101, 76, 31, 225, 1102, 60, 43, 225, 1102, 45, 31, 225, 1102, 63, 9, 225, 2, 170, 122, 224, 1001, 224, -486, 224, 4, 224, 102, 8, 223, 223, 101, 2, 224, 224, 1, 223, 224, 223, 1102, 29, 17, 224, 101, -493, 224, 224, 4, 224, 102, 8, 223, 223, 101, 1, 224, 224, 1, 223, 224, 223, 1102, 52, 54, 225, 1102, 27, 15, 225, 102, 26, 113, 224, 1001, 224, -1560, 224, 4, 224, 102, 8, 223, 223, 101, 7, 224, 224, 1, 223, 224, 223, 1002, 117, 81, 224, 101, -3645, 224, 224, 4, 224, 1002, 223, 8, 223, 101, 6, 224, 224, 1, 223, 224, 223, 4, 223, 99, 0, 0, 0, 677, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1105, 0, 99999, 1105, 227, 247, 1105, 1, 99999, 1005, 227, 99999, 1005, 0, 256, 1105, 1, 99999, 1106, 227, 99999, 1106, 0, 265, 1105, 1, 99999, 1006, 0, 99999, 1006, 227, 274, 1105, 1, 99999, 1105, 1, 280, 1105, 1, 99999, 1, 225, 225, 225, 1101, 294, 0, 0, 105, 1, 0, 1105, 1, 99999, 1106, 0, 300, 1105, 1, 99999, 1, 225, 225, 225, 1101, 314, 0, 0, 106, 0, 0, 1105, 1, 99999, 8, 226, 677, 224, 102, 2, 223, 223, 1005, 224, 329, 1001, 223, 1, 223, 1108, 677, 226, 224, 102, 2, 223, 223, 1006, 224, 344, 101, 1, 223, 223, 108, 677, 226, 224, 102, 2, 223, 223, 1006, 224, 359, 101, 1, 223, 223, 7, 677, 226, 224, 102, 2, 223, 223, 1005, 224, 374, 101, 1, 223, 223, 1007, 226, 677, 224, 102, 2, 223, 223, 1005, 224, 389, 101, 1, 223, 223, 8, 677, 677, 224, 102, 2, 223, 223, 1006, 224, 404, 1001, 223, 1, 223, 1007, 677, 677, 224, 1002, 223, 2, 223, 1006, 224, 419, 101, 1, 223, 223, 1108, 677, 677, 224, 1002, 223, 2, 223, 1005, 224, 434, 1001, 223, 1, 223, 1107, 226, 677, 224, 102, 2, 223, 223, 1005, 224, 449, 101, 1, 223, 223, 107, 226, 226, 224, 102, 2, 223, 223, 1006, 224, 464, 101, 1, 223, 223, 1108, 226, 677, 224, 1002, 223, 2, 223, 1005, 224, 479, 1001, 223, 1, 223, 7, 677, 677, 224, 102, 2, 223, 223, 1006, 224, 494, 1001, 223, 1, 223, 1107, 677, 226, 224, 102, 2, 223, 223, 1005, 224, 509, 101, 1, 223, 223, 107, 677, 677, 224, 1002, 223, 2, 223, 1006, 224, 524, 101, 1, 223, 223, 1008, 677, 677, 224, 1002, 223, 2, 223, 1006, 224, 539, 101, 1, 223, 223, 7, 226, 677, 224, 1002, 223, 2, 223, 1005, 224, 554, 101, 1, 223, 223, 108, 226, 226, 224, 1002, 223, 2, 223, 1006, 224, 569, 101, 1, 223, 223, 1008, 226, 677, 224, 102, 2, 223, 223, 1005, 224, 584, 101, 1, 223, 223, 8, 677, 226, 224, 1002, 223, 2, 223, 1005, 224, 599, 101, 1, 223, 223, 1007, 226, 226, 224, 1002, 223, 2, 223, 1005, 224, 614, 101, 1, 223, 223, 1107, 226, 226, 224, 1002, 223, 2, 223, 1006, 224, 629, 101, 1, 223, 223, 107, 677, 226, 224, 1002, 223, 2, 223, 1005, 224, 644, 1001, 223, 1, 223, 1008, 226, 226, 224, 1002, 223, 2, 223, 1006, 224, 659, 101, 1, 223, 223, 108, 677, 677, 224, 1002, 223, 2, 223, 1005, 224, 674, 1001, 223, 1, 223, 4, 223, 99, 226 ] # Interpretter spec: # Op-code width: 2 # ABCDE # A: Mode of 3rd parameter # B: Mode of 2rd parameter # C: Mode of 1st parameter # DE: 2-digit op-code # # Not every op-code has the same arity. # # Parameter modes: # - positional: index of memory. 0 # - immediate: raw value. 1 # Assert that you never attempt to write to an "immediate value" # Parameter modes POS = '0' # positional parameter mode VAL = '1' # immediate parameter mode # Pasted from day-2.py # interpretter :: Int -> [Int] -> [Int] -> IO () def interpret(i, x, argv=[], outs=[]): """Values in `argv` will be applied to any `input` fields.""" # The widest op-code we'll see is 3 + 2 = 5 for either addition or # multiplication since each of those is a 3-arity function with a two-digit # op-code. instruction = '{:05d}'.format(x[i]) op = instruction[-2:] if op == '01': a, b, out = x[i + 1], x[i + 2], x[i + 3] mode_a, mode_b, mode_out = instruction[2], instruction[1], instruction[ 0] a = a if mode_a == VAL else x[a] b = b if mode_b == VAL else x[b] assert mode_out == POS x[out] = a + b return interpret(i + 4, x, argv=argv, outs=outs) elif op == '02': a, b, out = x[i + 1], x[i + 2], x[i + 3] mode_a, mode_b, mode_out = instruction[2], instruction[1], instruction[ 0] a = a if mode_a == VAL else x[a] b = b if mode_b == VAL else x[b] assert mode_out == POS x[out] = a * b return interpret(i + 4, x, argv=argv, outs=outs) # input elif op == '03': a = x[i + 1] mode_a = instruction[2] assert mode_a == POS # What's the pythonic way to defensively get this value? if len(argv) and argv[0] is not None: x[a] = argv[0] return interpret(i + 2, x, argv=argv[1:], outs=outs) elif len(outs) and outs[-1] is not None: x[a] = outs[-1] return interpret(i + 2, x, argv=argv, outs=outs) else: # Here we want to block until the user applies input. This could be # done easily with message passing for something similar. x[a] = int(input('Enter: ')) return interpret(i + 2, x, argv=argv) # output elif op == '04': a = x[i + 1] mode_a = instruction[2] a = a if mode_a == VAL else x[a] outs.append(a) return interpret(i + 2, x, argv=argv, outs=outs) # jump-if-true elif op == '05': a, b = x[i + 1], x[i + 2] mode_a, mode_b = instruction[2], instruction[1] a = a if mode_a == VAL else x[a] b = b if mode_b == VAL else x[b] if a != 0: return interpret(b, x, argv=argv, outs=outs) else: return interpret(i + 3, x, argv=argv, outs=outs) # jump-if-false elif op == '06': a, b = x[i + 1], x[i + 2] mode_a, mode_b = instruction[2], instruction[1] a = a if mode_a == VAL else x[a] b = b if mode_b == VAL else x[b] if a == 0: return interpret(b, x, argv=argv, outs=outs) else: return interpret(i + 3, x, argv=argv, outs=outs) pass # less than elif op == '07': a, b, out = x[i + 1], x[i + 2], x[i + 3] mode_a, mode_b, mode_out = instruction[2], instruction[1], instruction[ 0] a = a if mode_a == VAL else x[a] b = b if mode_b == VAL else x[b] assert mode_out == POS if a < b: x[out] = 1 else: x[out] = 0 return interpret(i + 4, x, argv=argv, outs=outs) # equals elif op == '08': a, b, out = x[i + 1], x[i + 2], x[i + 3] mode_a, mode_b, mode_out = instruction[2], instruction[1], instruction[ 0] a = a if mode_a == VAL else x[a] b = b if mode_b == VAL else x[b] assert mode_out == POS if a == b: x[out] = 1 else: x[out] = 0 return interpret(i + 4, x, argv=argv, outs=outs) elif op == '99': return x[0] else: raise Exception('Unsupported opcode: {}.'.format(op))