Add problem 2600: K Items With the Maximum Sum

Author: EFanZh

src/lib.rs

@@ -1923,6 +1923,7 @@ pub mod problem_2594_minimum_time_to_repair_cars;
 pub mod problem_2595_number_of_even_and_odd_bits;
 pub mod problem_2596_check_knight_tour_configuration;
 pub mod problem_2598_smallest_missing_non_negative_integer_after_operations;
+pub mod problem_2600_k_items_with_the_maximum_sum;
 
 #[cfg(test)]
 mod test_utilities;

src/problem_2600_k_items_with_the_maximum_sum/mod.rs

@@ -0,0 +1,21 @@
+pub mod modular_arithmetic;
+
+pub trait Solution {
+    fn k_items_with_maximum_sum(num_ones: i32, num_zeros: i32, num_neg_ones: i32, k: i32) -> i32;
+}
+
+#[cfg(test)]
+mod tests {
+    use super::Solution;
+
+    pub fn run<S: Solution>() {
+        let test_cases = [((3, 2, 0, 2), 2), ((3, 2, 0, 4), 3), ((6, 6, 6, 13), 5)];
+
+        for ((num_ones, num_zeros, num_neg_ones, k), expected) in test_cases {
+            assert_eq!(
+                S::k_items_with_maximum_sum(num_ones, num_zeros, num_neg_ones, k),
+                expected,
+            );
+        }
+    }
+}

src/problem_2600_k_items_with_the_maximum_sum/modular_arithmetic.rs

@@ -0,0 +1,31 @@
+pub struct Solution;
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl Solution {
+    pub fn helper(num_ones: u32, num_zeros: u32, k: u32) -> u32 {
+        k.min(num_ones) - k.saturating_sub(num_ones + num_zeros)
+    }
+
+    pub fn k_items_with_maximum_sum(num_ones: i32, num_zeros: i32, num_neg_ones: i32, k: i32) -> i32 {
+        _ = num_neg_ones;
+
+        Self::helper(num_ones as _, num_zeros as _, k as _) as _
+    }
+}
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl super::Solution for Solution {
+    fn k_items_with_maximum_sum(num_ones: i32, num_zeros: i32, num_neg_ones: i32, k: i32) -> i32 {
+        Self::k_items_with_maximum_sum(num_ones, num_zeros, num_neg_ones, k)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    #[test]
+    fn test_solution() {
+        super::super::tests::run::<super::Solution>();
+    }
+}