day07 task 2

src/tasks/day07.rs

@@ -9,6 +9,7 @@ pub fn run() {
         .map(|s| s.parse::<i32>().unwrap())
         .collect();
     task1(prog_code.clone());
+    task2(prog_code.clone());
 }
 
 fn task1(input: Vec<i32>) {
@@ -26,3 +27,37 @@ fn task1(input: Vec<i32>) {
         .max();
     println!("Task 1: best signal is {}", result.unwrap());
 }
+
+fn task2(input: Vec<i32>) {
+    let phases = vec![5, 6, 7, 8, 9];
+    let result = phases
+        .iter()
+        .permutations(phases.len())
+        .map(|perm| {
+            let mut machines: Vec<IntCodeComputer> = perm
+                .iter()
+                .map(|phase| {
+                    let mut comp = IntCodeComputer::new(vec![**phase], input.clone());
+                    comp.run_until_input_empty();
+                    comp
+                })
+                .collect();
+            let mut m = 0;
+            let mut signal = 0;
+            let mut done = false;
+            while !done {
+                let mach = &mut machines[m];
+                mach.set_input(&vec![signal]);
+                let machine_done = mach.run_until_input_empty();
+                signal = *mach.get_output().last().unwrap();
+                mach.clear_output();
+                if machine_done && m == 4 {
+                    done = true;
+                }
+                m = (m + 1) % machines.len();
+            }
+            signal
+        })
+        .max();
+    println!("Task 2: best signal with looping is {}", result.unwrap());
+}