LC238——移动零

37 阅读1分钟

移动零

给定一个数组 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元素

  • 然后依次交换

  • 不能跳针

  • 错误的想法,这种无法覆盖[1,0]

public class MoveZeroesClass283 {
    public static void moveZeroes(int[] nums) {
        if (nums == null || nums.length == 1) return;

        int zIndex = 0;//0指针
        int nzIndex = 0;//非0指针

        while (zIndex < nums.length && nzIndex < nums.length) {
            if (nums[zIndex] != 0) {
                zIndex++;
                continue;
            }
            if (nums[nzIndex] == 0) {
                nzIndex++;
                continue;
            }
            //交换
            int temp = nums[zIndex];
            nums[zIndex] = nums[nzIndex];
            nums[nzIndex] = temp;
//            zIndex = nzIndex; //不能跳针
        }

        return;
    }
}
  • 参考快排想法,将0作为中间节点,遍历一遍,时间复杂度O(n)
public class MoveZeroClass {
    public void moveZeroes(int[] nums) {
        if (nums == null || nums.length == 1) return;

        int zIndex = 0;//0指针
        int nzIndex = 0;//非0指针

        //参考快排想法,将0作为中间节点
        while (zIndex < nums.length ) {
            if (nums[zIndex] != 0) {
                //交换
                int temp = nums[zIndex];
                nums[zIndex] = nums[nzIndex];
                nums[nzIndex++] = temp;
            }
            zIndex++;
        }
    }

    public static void main(String[] args) {
        int[] nums = {5,1,3,12,0};
        MoveZeroClass solution = new MoveZeroClass();
        solution.moveZeroes(nums);
        for (int i = 0 ; i < nums.length; i++) System.out.println(nums[i]);
    }
}

  • 更容易的实现,遍历两边,时间复杂度依然O(n)
//简单的实现
public void moveZeroes(int[] nums) {
    if (nums == null || nums.length == 1) return;

    int j = 0;//非0指针
    //将所有j指向所有非0元素
    for (int i = 0; i < nums.length; i++) {
        if (nums[i] != 0) {
            nums[j++] = nums[i];
        }
    }

    //将剩下的元素赋值为0元素
    for (int i = j; i < nums.length; i++) {
        nums[i] = 0;
    }
}