题目描述

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

示例:

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

说明:

1. 必须在原数组上操作，不能拷贝额外的数组。
2. 尽量减少操作次数。

https://leetcode-cn.com/problems/move-zeroes/

解法1 – 选择排序思想O(n^2)

这道题比较简单，一开始我想的是用选择排序的思想，如果nums[i] == 0，我们就从j = [i+1, |nums|)中找第一个不为0的元素交换到nums[i]，然后i++。这种做法是可行的，但是时间复杂度为O(n^2)，太蠢了。

class Solution {
    public void moveZeroes(int[] nums) {
        for (int i = 0; i < nums.length - 1; i++) {
            if (nums[i] == 0) {
                for (int j = i + 1; j < nums.length; j++) {
                    if (nums[j] != 0) {
                        int t = nums[i];
                        nums[i] = nums[j];
                        nums[j] = t;
                        break;
                    }
                }
            }
        }
    }
}

解法2 – 双指针O(n)

解法2采用双指针法，首先我们用header指针指向数组第0个元素，然后使用指针i遍历nums数组每个元素，当nums[i] != 0时，我们将nums[i]覆盖到数组头部，即nums[header++] = nums[i]。当i指针遍历到数组末尾，header之前所有的元素都是非0元素，我们将header指向位置之后的所有元素用0填充就是答案。

时间复杂度O(n)，空间复杂度O(1)。全部代码如下：

class Solution {
    public void moveZeroes(int[] nums) {
        int header = 0;
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] != 0) {
                nums[header++] = nums[i];
            }
        }
        while (header < nums.length) {
            nums[header++] = 0;
        }
    }
}