| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455 |
- #!/usr/bin/env python3
- import sys
- from itertools import permutations
- def run(p, inputs):
- pc = 0
- while p[pc] != 99:
- modes, opcode = divmod(p[pc], 100)
- def modeswitch(offset):
- value = p[pc + offset]
- mode = modes // (10 ** (offset - 1)) % 10
- return value if mode else p[value]
- if opcode in (1, 2):
- a = modeswitch(1)
- b = modeswitch(2)
- out = p[pc + 3]
- p[out] = a + b if opcode == 1 else a * b
- pc += 4
- elif opcode == 3:
- address = p[pc + 1]
- p[address] = inputs.pop()
- pc += 2
- elif opcode == 4:
- inputs = yield modeswitch(1)
- pc += 2
- elif opcode == 5:
- pc = modeswitch(2) if modeswitch(1) else pc + 3
- elif opcode == 6:
- pc = modeswitch(2) if not modeswitch(1) else pc + 3
- elif opcode in (7, 8):
- a = modeswitch(1)
- b = modeswitch(2)
- out = p[pc + 3]
- p[out] = int(a < b if opcode == 7 else a == b)
- pc += 4
- def amplify(p, phases):
- amps = []
- signal = 0
- for phase in phases:
- amp = run(list(p), [signal, phase])
- amps.append(amp)
- signal = next(amp)
- try:
- while True:
- for amp in amps:
- signal = amp.send([signal])
- except StopIteration:
- return signal
- program = list(map(int, sys.stdin.read().split(',')))
- print(max(amplify(program, phases) for phases in permutations(range(5))))
- print(max(amplify(program, phases) for phases in permutations(range(5, 10))))
|
