代码挑战:迁移0元素|Rust
Given an array nums
, write a function to move all 0
's to the end of it while maintaining the relative order of the non-zero elements.
Example:
Input: [0,1,0,3,12] Output: [1,3,12,0,0]
Note:
You must do this in-place without making a copy of the array.
Minimize the total number of operations.
方案:
impl Solution {
pub fn move_zeroes(nums: &mut Vec<i32>) {
let mut index = 0;
let len = nums.len();
for i in 0..len {
if nums[i] != 0 {
nums[index] = nums[i];
index +=1;
}
}
while index < len {
nums[index] = 0;
index +=1;
}
}
}
测试案例:
#[test]
fn test1() {
let mut vec: Vec<i32> = vec![0, 1, 0, 3, 12];
let expected = vec![1, 3, 12, 0, 0];
Solution::move_zeroes(&mut vec);
let len = vec.len();
for i in 0..len {
assert_eq!(vec[i], expected[i]);
}
}
#[test]
fn test2() {
let mut vec: Vec<i32> = vec![1, 9, 8, 4, 0, 0, 2, 7, 0, 6, 0, 9];
let expected = vec![1, 9, 8, 4, 2, 7, 6, 9, 0, 0, 0, 0];
let len = vec.len();
Solution::move_zeroes(&mut vec);
for i in 0..len {
assert_eq!(vec[i], expected[i]);
}
}
- 点赞
- 收藏
- 关注作者
评论(0)