2019 day7 part 2

2019/day7/day7.py

@@ -5,18 +5,23 @@ from pathlib import Path
 # installed module https://stackoverflow.com/a/50194143
 sys.path.append(str(Path(__file__).absolute().parent.parent / "intcode"))
 
-import itertools
-from intcode import interpret_intcode
+from itertools import cycle, permutations
+from intcode import interpret_intcode, Interpreter
 
 
 def main(inp):
     mem = list(map(int, inp.readline().rstrip().split(",")))
     max_ret = 0
-    for seq in itertools.permutations([0, 1, 2, 3, 4], 5):
+    for seq in permutations([0, 1, 2, 3, 4], 5):
         ret = amplifiers(mem, list(seq))
-        #print(ret)
         max_ret = max(ret[0], max_ret)
-    print(max_ret)
+    print("Part 1: ", max_ret)
+
+    max_ret = 0
+    for seq in permutations((5, 6, 7, 8, 9)):
+        ret = part2(mem, list(seq))
+        max_ret = max(max_ret, ret)
+    print("Part 2: ", max_ret)
 
 
 def amplifiers(program, sequence):
@@ -27,27 +32,65 @@ def amplifiers(program, sequence):
     ret = interpret_intcode(program[::], [sequence.pop(0), ret.pop()])
     return ret
 
+
+def part2(program, sequence):
+    amplifiers = [Interpreter(program[::], [sequence.pop(0)]) for _ in range(5)]
+    it = cycle(enumerate(amplifiers))
+
+    id_, amp = next(it)
+    inp = 0
+    max_signal = 0
+    while True:
+        max_signal = max(max_signal, inp)
+        amp.stdin.append(inp)
+        amp.interpret()
+        out = amp.stdout
+        if amp.halted:
+            break
+        next_id, next_amp = next(it)
+        inp = out.pop(0)
+        amp = next_amp
+        id_= next_id
+    return max_signal
+
+
 def tests():
     program = [3, 15, 3, 16, 1002, 16, 10, 16, 1, 16, 15, 15, 4, 15, 99, 0, 0]
     sequence = [4, 3, 2, 1, 0]
     res = amplifiers(program, sequence)
     assert res == [43210]
 
-    program = [3,23,3,24,1002,24,10,24,1002,23,-1,23, 101,5,23,23,1,24,23,23,4,23,99,0,0]
+    program = [3,23,3,24,1002,24,10,24,1002,23,-1,23,
+               101,5,23,23,1,24,23,23,4,23,99,0,0]
     sequence = [0,1,2,3,4]
     res = amplifiers(program, sequence)
     assert res == [54321]
 
 
-    program = [3,31,3,32,1002,32,10,32,1001,31,-2,31,1007,31,0,33, 1002,33,7,33,1,33,31,31,1,32,31,31,4,31,99,0,0,0]
+    program = [3,31,3,32,1002,32,10,32,1001,31,-2,31,1007,31,0,33,
+               1002,33,7,33,1,33,31,31,1,32,31,31,4,31,99,0,0,0]
     sequence = [1,0,4,3,2]
     res = amplifiers(program, sequence)
     assert res == [65210]
 
-    print("All tests passed")
+
+def tests2():
+    program = [3,26,1001,26,-4,26,3,27,1002,27,2,27,1,27,26,
+               27,4,27,1001,28,-1,28,1005,28,6,99,0,0,5]
+    sequence = [9,8,7,6,5]
+    assert part2(program, sequence) == 139629729
+
+
+    program = [3,52,1001,52,-5,52,3,53,1,52,56,54,1007,54,5,55,1005,55,26,1001,54,
+               -5,54,1105,1,12,1,53,54,53,1008,54,0,55,1001,55,1,55,2,53,55,53,4,
+               53,1001,56,-1,56,1005,56,6,99,0,0,0,0,10]
+    sequence = [9,7,8,5,6]
+    assert part2(program, sequence) == 18216
+
 
 if __name__ == "__main__":
     import fileinput
-    main(fileinput.input())
     tests()
+    tests2()
+    main(fileinput.input())