给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
我的思路是用一个队列保存0出现的位置,然后当遍历到的数字不为0,则可以出队得到第一个为0的位置,然后交换值,再把当前位置入队,因为当前位置已经成为新的0了。
from collections import deque
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
z = deque()
for i, n in enumerate(nums):
if n != 0:
if len(z) != 0:
a = z.popleft()
nums[a] = n
nums[i] = 0
z.append(i)
else:
z.append(i)
我的这个方法,空间和时间复杂度都是O(N)。时间复杂度肯定是不能减少了,空间复杂度却可以减小为1。我也是看了题解才知道第二种方法。
采用双指针法,快指针变量整个列表,慢指针记录最后一个非零位置之后的一个位置。也就是说慢指针指向整个列表第一个0的位置。
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
i = j = 0
while i < len(nums):
if nums[i] != 0:
nums[j] = nums[i]
j += 1
i += 1
while j < i:
nums[j] =0
j += 1