Add problem 3133: Minimum Array End

Author: EFanZh

src/lib.rs

@@ -2099,6 +2099,7 @@ 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_3131_find_the_integer_added_to_array_i;
+pub mod problem_3133_minimum_array_end;
 pub mod problem_3350_adjacent_increasing_subarrays_detection_ii;
 
 #[cfg(test)]

src/problem_3133_minimum_array_end/bit_manipulation.rs

@@ -0,0 +1,39 @@
+pub struct Solution;
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl Solution {
+    pub fn min_end(n: i32, x: i32) -> i64 {
+        let mut extra = u64::from(n.cast_unsigned() - 1);
+        let mut result = u64::from(x.cast_unsigned());
+        let mut processed = 0;
+
+        while extra != 0 {
+            processed += (result >> processed).trailing_ones();
+
+            let zeroes = (result >> processed).trailing_zeros();
+
+            result |= (extra & u64::unbounded_shl(1, zeroes).wrapping_sub(1)) << processed;
+            extra = u64::unbounded_shr(extra, zeroes);
+            processed += zeroes;
+        }
+
+        result.cast_signed()
+    }
+}
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl super::Solution for Solution {
+    fn min_end(n: i32, x: i32) -> i64 {
+        Self::min_end(n, x)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    #[test]
+    fn test_solution() {
+        super::super::tests::run::<super::Solution>();
+    }
+}

src/problem_3133_minimum_array_end/mod.rs

@@ -0,0 +1,18 @@
+pub mod bit_manipulation;
+
+pub trait Solution {
+    fn min_end(n: i32, x: i32) -> i64;
+}
+
+#[cfg(test)]
+mod tests {
+    use super::Solution;
+
+    pub fn run<S: Solution>() {
+        let test_cases = [((3, 4), 6), ((2, 7), 15), ((2, 2), 3)];
+
+        for ((n, x), expected) in test_cases {
+            assert_eq!(S::min_end(n, x), expected);
+        }
+    }
+}