给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
说明:必须在原数组上操作,不能拷贝额外的数组。
尽量减少操作次数。
这道题我的思路是:首先用一个变量a记录非零值的个数,遍历整个列表,遇到非零值就将其放到第a个位置上,这个非零值的位置上放0,然后a往后挪一位。这样列表开始的数值都是非零值,后面的数值都为0。时间复杂度为O(n),空间复杂度为O(1)。
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
a = 0
for i in range(len(nums)):
if nums[i] != 0:
if a != i:
nums[a] = nums[i]
nums[i] = 0
a = a+1
return nums
后来看了官方解法,学到“双指针”这个叫法,即用一个左指针指向处理好序列的尾部,也就是非零数的尾部;右指针指向待处理序列的头部。
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
a = 0
for i in range(len(nums)):
if nums[i] != 0:
if a != i:
nums[a] = nums[i]
nums[i] = 0
a = a+1
return nums