4.移动零

60 阅读1分钟

题目链接

给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。

请注意 ,必须在不复制数组的情况下原地对数组进行操作。

示例 1:

输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]

示例 2:

输入: nums = [0]
输出: [0]

题解1 删除再添加

思路

可以先统计出0的个数,并在统计时删除0。最后在末位补0即可。

要注意如果从头遍历,使用 splice 删除时会打乱遍历顺序,所以要从后遍历。

代码

function moveZeroes(nums: number[]): void {
    let count: number = 0;

    for (let i = nums.length - 1; i >= 0; i--) {
        if (nums[i] === 0) {
            nums.splice(i, 1);
            count++;
        }
    }

    for (let i = 0; i < count; i++) {
        nums.push(0);
    }
};

时空复杂度分析

时间复杂度:一次遍历加上答案常数,所以为 O(n)

空间复杂度:只使用了1个变量,所以是 O(1)

题解2 双指针

思路

这题的难点在于原地对数组进行操作,所以上面的增删数组是不允许的。

既然是原地,那么肯定是使用 swap 来交换两个数的位置。而首先我们要找到这两个数,于是需要有两个指针,所以这题可以用双指针来做。

指针 i 和指针 j 从头遍历,只要 i 不是 0 就进行交换,并且 j 指针往前走。

代码

function moveZeroes(nums: number[]): void {
    let j: number = 0;
    for (let i = 0; i < nums.length; i++) {
        if (nums[i] !== 0) {
            const temp: number = nums[j];
            nums[j++] = nums[i];
            nums[i] = temp;
        }
    }
};

时空复杂度分析

时间复杂度:一次遍历 O(n)

空间复杂度:O(1)