Add problem 3200: Maximum Height of a Triangle

Author: EFanZh

src/lib.rs

@@ -2133,6 +2133,7 @@ pub mod problem_3192_minimum_operations_to_make_binary_array_elements_equal_to_o
 pub mod problem_3194_minimum_average_of_smallest_and_largest_elements;
 pub mod problem_3195_find_the_minimum_area_to_cover_all_ones_i;
 pub mod problem_3196_maximize_total_cost_of_alternating_subarrays;
+pub mod problem_3200_maximum_height_of_a_triangle;
 pub mod problem_3350_adjacent_increasing_subarrays_detection_ii;
 
 #[cfg(test)]

src/problem_3200_maximum_height_of_a_triangle/mathematical.rs

@@ -0,0 +1,50 @@
+pub struct Solution;
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl Solution {
+    fn max_even_rows(x: u32) -> u32 {
+        x.isqrt()
+    }
+
+    fn max_odd_rows(x: u32) -> u32 {
+        ((x * 4 + 1).isqrt() - 1) / 2
+    }
+
+    fn get_max_rows(even_rows: u32, odd_rows: u32) -> u32 {
+        if odd_rows < even_rows {
+            odd_rows * 2 + 1
+        } else {
+            even_rows * 2
+        }
+    }
+
+    pub fn max_height_of_triangle(red: i32, blue: i32) -> i32 {
+        let red = red.cast_unsigned();
+        let blue = blue.cast_unsigned();
+        let red_even_rows = Self::max_even_rows(red);
+        let red_odd_rows = Self::max_odd_rows(red);
+        let blue_even_rows = Self::max_even_rows(blue);
+        let blue_odd_rows = Self::max_odd_rows(blue);
+
+        Self::get_max_rows(red_even_rows, blue_odd_rows)
+            .max(Self::get_max_rows(blue_even_rows, red_odd_rows))
+            .cast_signed()
+    }
+}
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl super::Solution for Solution {
+    fn max_height_of_triangle(red: i32, blue: i32) -> i32 {
+        Self::max_height_of_triangle(red, blue)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    #[test]
+    fn test_solution() {
+        super::super::tests::run::<super::Solution>();
+    }
+}

src/problem_3200_maximum_height_of_a_triangle/mod.rs

@@ -0,0 +1,18 @@
+pub mod mathematical;
+
+pub trait Solution {
+    fn max_height_of_triangle(red: i32, blue: i32) -> i32;
+}
+
+#[cfg(test)]
+mod tests {
+    use super::Solution;
+
+    pub fn run<S: Solution>() {
+        let test_cases = [((2, 4), 3), ((2, 1), 2), ((4, 9), 4)];
+
+        for ((red, blue), expected) in test_cases {
+            assert_eq!(S::max_height_of_triangle(red, blue), expected);
+        }
+    }
+}