LeetCode 283.移动零

137 阅读2分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

题目:给定一个数组nums,要求将数组中的所有0移动到数组的最后,并且要求保持数组中所有非0元素的相对顺序。此处不允许复制数组,也就是要求空间复杂度为O(1)O(1)

解题思路

本题虽然说是个简单题,但要求的逻辑能力还是不低的。

首先我们可以使用索引定位法,题目要求我们将数组中的零元素全部移动到最后,那么我们在遍历数组的过程中忽略零元素,每次遍历将非零元素的位置确定,最后对数组尾部元素赋值为0即可。

具体来说,我们使用一个变量来记录当前已确定的数组尾部索引,当数组每次遍历发现元素非零时,将该元素赋值给已确定索引位置,之后更新索引位置。当数组遍历完毕之后,根据已确定索引和数组总长进行判断,如果小于则直接赋值剩下的即可,可得代码如下:

public void moveZeroes(int[] nums) {
    int left = 0, right = 0;
    while(right<nums.length){
        if(nums[right]!=0){
            swap(nums, right, left);
            left++;
        }
        right++;
    }
}

public void swap(int[] nums, int i, int j){
    int temp = nums[i];
    nums[i] = nums[j];
    nums[j] = temp;
}

时间复杂度为O(n)O(n),空间复杂度为O(1)O(1)

上面的方法虽然解决了问题,但直接修改数组中元素还是不合适。如果我们此时遇到更复杂的数据结构,例如一个用户,我们需要按照用户的类型作此操作,但每个用户显然不同,因此这种方式就失效了。

对于不直接修改数组中元素,我们可以使用双指针法,其中左指针指向已确定数组的尾部,而右指针指向待确定数组首部。这样每当找到一个元素不等于0,则直接将右指针和左指针元素互换,更新两个指针即可。如果等于0则只更新右指针。可得代码如下:

public void moveZeroes(int[] nums) {
    int left = 0, right = 0;
    while(right<nums.length){
        if(nums[right]!=0){
            swap(nums, right, left);
            left++;
        }
        right++;
    }
}

上述代码时间复杂度为O(n)O(n),空间复杂度为O(1)O(1)。但实际测试明显比上面的代码满慢。