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