力扣:移动零

news2024/12/1 2:30:33

文章目录

需求
分析
优化下
双指针
结尾

在这里插入图片描述

需求

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

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

示例 2:
输入: nums = [0]
输出: [0]

提示:
1 <= nums.length <= 104
-231 <= nums[i] <= 231 - 1

进阶：你能尽量减少完成的操作次数吗？

分析

就是将非0元素全部移动到前面, 然后将0 移动到后面, 那我们可以遍历数组, 遍历到过程中, 维护一个变量, 该变量指向数组实际保存的非0元素的位置, 然后在遍历的过程中, 非0就移动到变量的位置,再将变量++, 变量完后将变量后面的填充0;

public void moveZeroes(int[] nums) {
    int length = nums.length;
    int index = 0;
    for( int i = 0; i < length; i++ ){
        if( nums[i] != 0 ){
            nums[index++] = nums[i];
        }
    }
    //  后面的填充0
    for( int i = index; i < length; i++ ){
        nums[i] = 0;
    }
}

运行后结果如下:

在这里插入图片描述

优化下

提示说 1 <= nums.length <= 104, 当数组长度是1的时候,就没有任何比较移动的必要了, 直接返回即可:

public void moveZeroes(int[] nums) {
    int length = nums.length;
    if( length == 1 ){
        return;
    }
    int index = 0;
    for( int i = 0; i < length; i++ ){
        if( nums[i] != 0 ){
            nums[index++] = nums[i];
        }
    }
    //  后面的填充0
    for( int i = index; i < length; i++ ){
        nums[i] = 0;
    }
}

运行结果:

在这里插入图片描述

双指针

上面的两种做法都是遍历了两遍数组, 那么能不能只遍历一遍数组呢, 试试双指针吧:

public void moveZeroes(int[] nums) {
    int length = nums.length;
    if( length == 1 ){
        return;
    }
    int slow = 0; // 指向下一个非零元素的位置
    // 遍历数组
    for ( int fast = 0; fast < nums.length; fast++ ) {
        if (nums[fast] != 0) {
            // 交换非零元素到 slow
            int temp = nums[slow];
            nums[slow] = nums[fast];
            nums[fast] = temp;
            
            // 更新 slow
            slow++;
        }
    }
}