1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37
#![allow(dead_code)] pub fn move_zeroes(nums: &mut Vec<i32>) { let mut i = 0; let mut j = 0; while i < nums.len() { if nums[i] != 0 { nums.swap(i, j); j += 1; } i += 1; } } #[cfg(test)] mod tests { use super::*; #[test] fn test1() { let mut nums = vec![0, 1, 0, 3, 12]; move_zeroes(&mut nums); assert_eq!(nums, vec![1, 3, 12, 0, 0]); let mut nums = vec![1]; move_zeroes(&mut nums); assert_eq!(nums, vec![1]); let mut nums = vec![1, 0]; move_zeroes(&mut nums); assert_eq!(nums, vec![1, 0]); let mut nums = vec![2, 1]; move_zeroes(&mut nums); assert_eq!(nums, vec![2, 1]); } }