Leetcode.283 移动零

题目链接

Leetcode.283 移动零 easy

题目描述

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

请注意 ，必须在不复制数组的情况下原地对数组进行操作。

示例 1:

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

示例 2:

输入: nums = [0]
输出: [0]

提示:

• $1<=nums.length<=10^4$
• $-2^{31}<=nums[i]<=2^{31}-1$

解法：双指针

用两个指针 i和 j，我们的操作保证 j之前的元素都是非0的（j初始为 0）。

遍历数组，当 nums[i] != 0时 并且 j < i时，执行交换操作，nums[j] = nums[i] , nums[i] = 0。这个操作就是把 0换到后面i去，把非0的数 nums[i]换到前面 j来。

当 i == j时交换是没有意义的。

时间复杂度： $O(n)$

C++代码：

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        int n = nums.size();
        int j = 0;
        for(int i = 0;i < n;i++){
            if(nums[i] != 0){
                if(i > j){
                    nums[j] = nums[i];
                    nums[i] = 0;
                }
                j++;
            }
        }        
    }
};

Python代码：

class Solution:
    def moveZeroes(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        n = len(nums)
        j = 0
        for i in range(n):
            if nums[i] != 0:
                if i > j:
                    nums[j] = nums[i]
                    nums[i] = 0
                j += 1