题目描述

给定一个数组 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;
        }
    }
}
pwrliang Algorithms, Array, Sort

Leave a Reply

Your email address will not be published. Required fields are marked *