intcode.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-

import sys


DEPTH_MAX = 5
PHASE_MIN = 0
PHASE_MAX = 4
BEST_RESULT = 0
BEST_CONFIG = []


def decode_opcode(value):
	op = value % 100
	value = value // 100
	
	argmode = []
	for i in range(3):
		argmode.append(value % 10)
		value = value // 10

	return op, argmode


def get_val(ints, position, mode):
	if mode == 0:
		return ints[position]
	elif mode == 1:
		return position
	else:
		print("What the fuck", file=sys.stderr)
		exit(1)


def run_script(ints, input):
	ints = ints.copy()
	output = []

	pos = 0
	while True:
		op, argmode = decode_opcode(ints[pos])

		if op == 1:
			arg1 = ints[pos+1]
			arg2 = ints[pos+2]
			arg3 = ints[pos+3]
			pos += 4
			
			ints[arg3] = get_val(ints, arg1, argmode[0]) + get_val(ints, arg2, argmode[1])
		elif op == 2:
			arg1 = ints[pos+1]
			arg2 = ints[pos+2]
			arg3 = ints[pos+3]
			pos += 4
			
			ints[arg3] = get_val(ints, arg1, argmode[0]) * get_val(ints, arg2, argmode[1])
		elif op == 3:
			arg = ints[pos+1]
			pos += 2

			ints[arg] = input.pop(0)
		elif op == 4:
			arg = ints[pos+1]
			pos += 2

			output.append(get_val(ints, arg, argmode[0]))
			# print("Opcode 4: {}".format(get_val(ints, arg, argmode[0])))
		elif (op == 5) or (op == 6):
			arg1 = ints[pos+1]
			arg2 = ints[pos+2]
			
			val = get_val(ints, arg1, argmode[0])
			if ((op == 5) and (val != 0)) or ((op == 6) and (val == 0)):
				pos = get_val(ints, arg2, argmode[1])
			else:
				pos += 3
		elif (op == 7) or (op == 8):
			arg1 = ints[pos+1]
			arg2 = ints[pos+2]
			arg3 = ints[pos+3]
			pos += 4

			val1 = get_val(ints, arg1, argmode[0])
			val2 = get_val(ints, arg2, argmode[1])

			condition = ((op == 7) and (val1 < val2)) or ((op == 8) and (val1 == val2))
			ints[arg3] = 1 if condition else 0
		elif op == 99:
			break
		else:
			print("Unknown opcode \"{}\"".format(ints[pos]), file=sys.stderr)
			exit(1)
	
	return output


def str_to_ints(string):
	return list(map(lambda x: int(x), string.split(',')))


def ints_to_str(ints):
	return ",".join(list(map(lambda x: str(x), ints)))


def run(depth, script, input, history):
	global DEPTH_MAX
	global PHASE_MIN
	global PHASE_MAX
	global BEST_RESULT
	global BEST_CONFIG

	for phase in range(PHASE_MIN, PHASE_MAX+1):
		if phase in history: continue

		newhist = history.copy()
		newhist.append(phase)
		
		output = run_script(script, [phase,input])
		output = output[0]

		if depth < DEPTH_MAX:
			run(depth+1, script, output, newhist)
		else:
			print("{} ---> {}".format(ints_to_str(BEST_CONFIG), output))

			if output > BEST_RESULT:
				BEST_RESULT = output
				BEST_CONFIG = newhist


if __name__ == "__main__":
	script = str_to_ints(sys.stdin.read())
	run(1, script, 0, [])
	print("!! {} ---> {}".format(ints_to_str(BEST_CONFIG), BEST_RESULT))