力扣 - 283移动零

199 阅读1分钟

题目链接

题目详情

给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。 请注意 ,必须在不复制数组的情况下原地对数组进行操作。

示例 1:

输入:

nums = [0,1,0,3,12]

输出:

[1,3,12,0,0]

示例 2:

输入:

nums = [0]

输出:

[0]

提示:

1<=nums.length<=1041 <= nums.length <= 10^4
231<=nums[i]<=2311-2^{31} <= nums[i] <= 2^{31} - 1

这里提供两种解题思路,第一种较好理解。第二种则类似于快速排序,找一个枢轴在其左右分类。

解题思路1

第一种方法,可以用两个指针i,j,先将所有的不为0的数赋值给数组,j记录不为0的数的个数。当所有!=0的数全部移动到前面时,从j开始再往数组后面补0,直到达到数组长度即可。

AC代码

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        int j=0;
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]!=0)
            {
                nums[j++] = nums[i];
            }
        }
​
        for(int i=j;i<nums.size();i++)
        {
            nums[i] = 0;
        }
    }
};

解题思路2

第二种方法使用双指针,有点类似于快速排序的思想,L指针指向当前已经处理好的序列的尾部,R指针指向待处理序列的头部。

这里我们把 0 当做枢轴,若不等于 0则把数放到0的左边,把等于 0 的数放到其右边。这里,我们使用两个指针 i 和 j,只要 nums[i]!=0,我们就交换 nums[i] 和 nums[j]。

右指针不断向右移动,每次右指针指向非零数,则将左右指针对应的数交换,同时左指针右移。每次交换,都是将左指针的0与右指针的非零数交换,且非零数的相对顺序并未改变。

AC代码

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        int n = nums.size(), l = 0, r = 0;
        while (r < n) {
            if (nums[r]) {
                swap(nums[r], nums[l]);
                l++;
            }
            r++;
        }
    }
};