diff options
Diffstat (limited to 'other/remove_element.rs')
-rw-r--r-- | other/remove_element.rs | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/other/remove_element.rs b/other/remove_element.rs new file mode 100644 index 0000000..fec38b6 --- /dev/null +++ b/other/remove_element.rs @@ -0,0 +1,22 @@ +use std::convert::TryInto; +struct Solution; + +impl Solution { + pub fn remove_element(mut nums: Vec<i32>, val: i32) -> i32 { + let (mut idx, mut k) = (0, nums.len()); + while idx < k { + if nums[idx] == val { + nums[idx] = nums[k - 1]; + k -= 1; + } else { + idx += 1; + } + } + k.try_into().unwrap() + } +} + +fn main() { + let nums = vec![0,1,2,2,3,0,4,2]; + print!("{}", Solution::remove_element(nums, 2)); +} |