Create 07-mice-and-cheese.rs

xhofe · Jun 7, 2023 · 2061d22 · 2061d22
1 parent d0ca8fe
commit 2061d22
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/2023/06/07-mice-and-cheese.rs b/2023/06/07-mice-and-cheese.rs
@@ -0,0 +1,12 @@
+impl Solution {
+    pub fn mice_and_cheese(reward1: Vec<i32>, reward2: Vec<i32>, k: i32) -> i32 {
+        let mut diff = reward1
+            .iter()
+            .zip(reward2.iter())
+            .map(|(&x, &y)| (x, y, x - y))
+            .collect::<Vec<_>>();
+        diff.sort_unstable_by_key(|x| -x.2);
+        diff.iter().take(k as usize).map(|x| x.0).sum::<i32>()
+            + diff.iter().skip(k as usize).map(|x| x.1).sum::<i32>()
+    }
+}