Add problem 3127: Make a Square with the Same Color

Author: EFanZh

src/lib.rs

@@ -2097,6 +2097,7 @@ pub mod problem_3114_latest_time_you_can_obtain_after_replacing_characters;
 pub mod problem_3115_maximum_prime_difference;
 pub mod problem_3120_count_the_number_of_special_characters_i;
 pub mod problem_3121_count_the_number_of_special_characters_ii;
+pub mod problem_3127_make_a_square_with_the_same_color;
 pub mod problem_3350_adjacent_increasing_subarrays_detection_ii;
 
 #[cfg(test)]

src/problem_3127_make_a_square_with_the_same_color/brute_force.rs

@@ -0,0 +1,37 @@
+pub struct Solution;
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl Solution {
+    pub fn can_make_square(grid: Vec<Vec<char>>) -> bool {
+        let [[m00, m01, m02], [m10, m11, m12], [m20, m21, m22]] = <[_; 3]>::try_from(grid)
+            .ok()
+            .unwrap()
+            .map(|row| <[_; 3]>::try_from(row).ok().unwrap());
+
+        [
+            [m00, m01, m10, m11],
+            [m01, m02, m11, m12],
+            [m10, m11, m20, m21],
+            [m11, m12, m21, m22],
+        ]
+        .iter()
+        .any(|values| values.iter().filter(|&&c| c == 'B').count() != 2)
+    }
+}
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl super::Solution for Solution {
+    fn can_make_square(grid: Vec<Vec<char>>) -> bool {
+        Self::can_make_square(grid)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    #[test]
+    fn test_solution() {
+        super::super::tests::run::<super::Solution>();
+    }
+}

src/problem_3127_make_a_square_with_the_same_color/mod.rs

@@ -0,0 +1,26 @@
+pub mod brute_force;
+
+pub trait Solution {
+    fn can_make_square(grid: Vec<Vec<char>>) -> bool;
+}
+
+#[cfg(test)]
+mod tests {
+    use super::Solution;
+
+    pub fn run<S: Solution>() {
+        let test_cases = [
+            (["BWB", "BWW", "BWB"], true),
+            (["BWB", "WBW", "BWB"], false),
+            (["BWB", "BWW", "BWW"], true),
+            (["BBB", "BBB", "BBB"], true),
+        ];
+
+        for (board, expected) in test_cases {
+            assert_eq!(
+                S::can_make_square(board.iter().map(|row| row.chars().collect()).collect()),
+                expected,
+            );
+        }
+    }
+}