跳转至

移动零

约 174 个字 45 行代码 预计阅读时间 1 分钟

题目描述

 给定一个整数数组nums，将nums里的所有0移动到数组的右边，同时需要保持非零元素的相对顺序不变。

 例如:[0,1,0,4,12] ->[1,4,12,0,0]

思路

 题目要求把0移动到右边，其实就是将非零元素移动到数组的左边。定义双指针i和j.其中 i 用于遍历所有 nums 中的所有元素，找到非 0 元素，j 用于指向非零序列尾部的插入位置。当nums[i] != 0时，交换nums[i]和nums[j]的值即可。

代码

Rust
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
struct Solution;

impl Solution {
    pub fn move_zeroes(nums: &mut Vec<i32>) {
        let mut p0 = 0; // 指向的是非零序列尾部插入位置
        let mut p1 = 0; // 总是指向非0的位置
        while p0 < nums.len() {
            if nums[p0] != 0 {
                let t = nums[p0];
                nums[p0] = nums[p1];
                nums[p1] = t;
                p1 += 1;
            }
            p0 += 1;
        }
    }
}

 使用 Rust 语言特性，还可以继续简化代码：

Rust
 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
struct Solution;

impl Solution {
    pub fn move_zeroes(nums: &mut Vec<i32>) {
        let mut j: usize = 0; // 始终指向非零序列尾部的插入位置
        for i in 0..nums.len() {
            if nums[i] != 0 {
                (nums[i], nums[j]) = (nums[j], nums[i]);
                j += 1 // 插入位置右移
            }
        }
    }
}

/// # Move Zeroes
#[cfg(test)]
pub mod test_283 {
    use super::Solution;

    #[test]
    fn test_p283() {
        let mut arr = vec![0, 1, 0, 3, 12];
        Solution::move_zeroes(&mut arr);
        assert_eq!(vec![1, 3, 12, 0, 0], arr);
    }
}