Add problem 3033: Modify the Matrix

Author: EFanZh

src/lib.rs

@@ -2054,6 +2054,7 @@ pub mod problem_3021_alice_and_bob_playing_flower_game;
 pub mod problem_3024_type_of_triangle;
 pub mod problem_3026_maximum_good_subarray_sum;
 pub mod problem_3028_ant_on_the_boundary;
+pub mod problem_3033_modify_the_matrix;
 
 #[cfg(test)]
 mod test_utilities;

src/problem_3033_modify_the_matrix/iterative.rs

@@ -0,0 +1,40 @@
+pub struct Solution;
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl Solution {
+    pub fn modified_matrix(matrix: Vec<Vec<i32>>) -> Vec<Vec<i32>> {
+        let mut matrix = matrix;
+        let columns = matrix.first().map_or(0, Vec::len);
+
+        (0..columns).for_each(|column| {
+            let max = matrix.iter().fold(i32::MIN, |max, row| max.max(row[column]));
+
+            for row in &mut matrix {
+                let value = &mut row[column];
+
+                if *value == -1 {
+                    *value = max;
+                }
+            }
+        });
+
+        matrix
+    }
+}
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl super::Solution for Solution {
+    fn modified_matrix(matrix: Vec<Vec<i32>>) -> Vec<Vec<i32>> {
+        Self::modified_matrix(matrix)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    #[test]
+    fn test_solution() {
+        super::super::tests::run::<super::Solution>();
+    }
+}

src/problem_3033_modify_the_matrix/mod.rs

@@ -0,0 +1,25 @@
+pub mod iterative;
+
+pub trait Solution {
+    fn modified_matrix(matrix: Vec<Vec<i32>>) -> Vec<Vec<i32>>;
+}
+
+#[cfg(test)]
+mod tests {
+    use super::Solution;
+    use crate::test_utilities::Matrix;
+
+    pub fn run<S: Solution>() {
+        let test_cases = [
+            (
+                &[[1, 2, -1], [4, -1, 6], [7, 8, 9]] as &dyn Matrix<_>,
+                &[[1, 2, 9], [4, 8, 6], [7, 8, 9]] as &dyn Matrix<_>,
+            ),
+            (&[[3, -1], [5, 2]], &[[3, 2], [5, 2]]),
+        ];
+
+        for (matrix, expected) in test_cases {
+            assert_eq!(S::modified_matrix(matrix.to_vec()), expected);
+        }
+    }
+}