day07.py

#!/usr/bin/env python3

import sys
from itertools import permutations
from lib.intcode import IntcodeVM

# Open the first argument as input or use stdin if no arguments were given
fin = open(sys.argv[1]) if len(sys.argv) > 1 else sys.stdin

program = list(map(int, fin.read().split(',')))
vms = [IntcodeVM(program) for _ in range(5)]
max_signal = float('-inf')

for inputs in permutations(range(5), 5):
	out = [0]

	for vm, inp in zip(vms, inputs):
		out = vm.run([inp, *out])

	if out[0] > max_signal:
		max_signal = out[0]

print('Part 1:', max_signal)


max_signal = float('-inf')

for inputs in permutations(range(5, 10), 5):
	out = [0]

	for vm, inp in zip(vms, inputs):
		out = vm.run([inp, *out], n_out=1)

	last_signal = out[0]

	while all(vm.running for vm in vms):
		for i, vm in enumerate(vms):
			out = vm.run(out, n_out=1, resume=True)

			if not vm.running:
				break

			if i == 4:
				last_signal = out[0]

	if last_signal > max_signal:
		max_signal = last_signal

print('Part 2:', max_signal)