给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
遍历数组,每次遍历后pop出0,之后再在原数组上追加pop出的0即可
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
cnt = 0
nums2 = nums.copy()
for i in range(len(nums)):
if nums2[i] == 0:
nums.pop(i-cnt)
cnt +=1
nums += [0]*cnt
有点需要小注意的是pop后原数组会变小,遍历到最后会越界,所以可以copy一个新数组,遍历新数组,pop原数组
cnt用来计数,使pop后仍能找到正确的index
之后又想了个不用copy的,思路基本一致,代码如下
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
cnt = 0
ls = []
for i in range(len(nums)):
if nums[i] == 0:
ls += [i]
for i in ls:
nums.pop(i-cnt)
cnt += 1
nums += [0]*cnt