【数组&双指针】leetcode 283. 移动零【简单】

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

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

【分析】

方法一：两次遍历

创建两个指针i和j，第一次遍历的时候指针j用来记录当前都多少非零元素。即遍历的时候每遇到一个非零元素就将其往数组左边挪，第一次遍历完后，j指针的下标就指向了最后一个非零元素的下标。

第二次遍历的时候，起始位置从j开始到结束，将剩下的这段区域内的元素全部置为0。

时间复杂度：O(n)

空间复杂度：O(1)

class Solution(object):
    def moveZeroes(self, nums):
        """
        :type nums: List[int]
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        if not nums:
            return 0
        # 第一次遍历的时候，j指针记录非0的个数，只要是非0的统统都赋给nums[j]    
        j = 0
        for i in xrange(len(nums)):
            if nums[i]:
                nums[j] = nums[i]
                j += 1
        # 非0元素统计完了，剩下的都是0了
        # 所以第二次遍历把末尾的元素都赋为0即可
        for i in xrange(j,len(nums)):
            nums[i] = 0

【方法二: 一次遍历】重点学习一下！

这里参考了快速排序的思想，快速排序首先要确定一个待分割的元素做中间点x，然后把所有小于等于x的元素放到x左边，大于x的元素放到其右边。

这里我们可以使用0作为中间点，把不等于0的放到中间点的左边，等于0的放到其右边。

这里的中间点就是0本身，所以实现起来比快速排序简单很多，我们使用两个指针i和j，只要nums[i] != 0，我们就交换nums[i]和nums[j]。

时间复杂度：O(n)

空间复杂度：O(1)

class Solution:
    def moveZeroes(self, nums: List[int]) -> None:
        n = len(nums)
        left, right = 0, 0 
        while right < n:
            if nums[right] != 0:
                nums[left], nums[right] = nums[right], nums[left]
                left += 1
            right += 1

等同于：

class Solution(object):
    def moveZeroes(self, nums):
        """
        :type nums: List[int]
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        if not nums:
            return 0
        # 两个指针i和j
        j = 0
        for i in xrange(len(nums)):
            # 当前元素!=0，就把其交换到左边，等于0的交换到右边
            if nums[i]:
                nums[j],nums[i] = nums[i],nums[j]
                j += 1