Update to also do part 2

5/src/main.rs

@@ -57,27 +57,48 @@ fn sum_middle_page_nums(updates: Vec<Vec<&str>>) -> i64 {
         .sum()
 }
 
+fn correct_update(update: &mut Vec<&str>, before_index: usize) {
+    let elem = update.remove(before_index);
+    update.insert(0, elem);
+}
+
 fn main() -> anyhow::Result<()> {
     let data = get_data()?;
     let (rules, updates) = parse_rules_and_updates(&data);
     let mut correct_updates = Vec::new();
+    let mut corrected_updates = Vec::new();
 
-    'update: for update in updates {
+    'update: for mut update in updates {
-        for rule in &rules {
+        let mut fixed = false;
-            if let Some(_) = check_rule_compliance(&update, rule) {
+        let mut keep_fixing = true;
-                /*println!(
+
-                    "Rule issue: {:?} ({before_index:?}) {:?} ({after_index:?})",
+        while keep_fixing {
-                    update[before_index],
+            keep_fixing = false;
-                    update[after_index]
+            for rule in &rules {
-                );
+                if let Some((before_index, after_index)) = check_rule_compliance(&update, rule) {
-                println!("Elements: {update:?}");*/
+                    fixed = true;
-                continue 'update;
+                    keep_fixing = true;
+                    println!("Elements: {update:?}");
+                    println!(
+                        "Rule issue: {:?} ({before_index:?}) {:?} ({after_index:?})",
+                        update[before_index],
+                        update[after_index]
+                    );
+                    correct_update(&mut update, before_index);
+                    println!("Elements: {update:?}");
+                }
             }
         }
-        correct_updates.push(update);
+
+        if fixed {
+            corrected_updates.push(update);
+        } else {
+            correct_updates.push(update);
+        }
     }
-    
+   
     println!("{}", sum_middle_page_nums(correct_updates));
+    println!("{}", sum_middle_page_nums(corrected_updates));
     Ok(())
 }