leetcode-283.移动0

leetcode-283 题目要求

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

示例:

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

说明:

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

一种解法：

def moveZeros(nums, toLeft=False):
	"""
	@param toLeft: 是否将0移动到左边
	O(kn)
	"""
    if toLeft:
        length = len(nums)
        for i in range(length):
            if nums[i] != 0:
                for j in range(i+1, length):
                    if nums[j] == 0:
                        nums[j], nums[i] = nums[i], nums[j]
                        break
    else:
        length = len(nums)
        for i in range(length):
            if nums[i] == 0:
                for j in range(i+1, length):
                    if nums[j] != 0:
                        nums[j], nums[i] = nums[i], nums[j]
                        break

上面的解法太慢了，败给了91.2%的提交。
下面的解法比较好：

def moveZeros(nums, toLeft=False):
	"""
	O(n)
	"""
	length = len(nums)
	curr = 0
	for i in range(length):
		if nums[i] != 0:
			nums[curr] = nums[i]
			curr += 1
	for j in range(curr, length):
		nums[j] = 0