| 1234567891011121314151617181920212223242526272829303132333435363738 |
- #!/usr/bin/env python3
- import sys
- from intcode import read_program, run
- def run_nat(program):
- nics, messages = zip(*[(run(program), [i]) for i in range(50)])
- natpack = 0, 0
- for nic in nics:
- assert next(nic) is None
- while True:
- for nic, msg in zip(nics, messages):
- msg.append(-1)
- while msg:
- dst = nic.send(msg.pop(0))
- while dst is not None:
- pack = next(nic), next(nic)
- if dst == 255:
- natpack = pack
- else:
- messages[dst].extend(pack)
- dst = next(nic)
- if not any(messages):
- messages[0].extend(natpack)
- yield natpack[1]
- def find_repeated_natpack(program):
- seen = set()
- for y in run_nat(program):
- if y in seen:
- return y
- seen.add(y)
- program = read_program(sys.stdin)
- print(next(run_nat(program)))
- print(find_repeated_natpack(program))
|
