Create 05-apply-operations-to-an-array.rs

xhofe · Jun 6, 2023 · d884f93 · d884f93
1 parent 560c9e6
commit d884f93
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/2023/06/05-apply-operations-to-an-array.rs b/2023/06/05-apply-operations-to-an-array.rs
@@ -0,0 +1,13 @@
+impl Solution {
+    pub fn apply_operations(mut nums: Vec<i32>) -> Vec<i32> {
+        (0..nums.len() - 1).for_each(|i| {
+            if nums[i] == nums[i + 1] {
+                nums[i] *= 2;
+                nums[i + 1] = 0;
+            }
+        });
+        let (mut l, r): (Vec<_>, Vec<_>) = nums.into_iter().partition(|&x| x != 0);
+        l.extend(r);
+        l
+    }
+}