给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
思路:定义一个指针k,记录从0到k处的值都是已排好的非零数,遍历时遇到0跳过,遇到非零的数时将其交换至k位置,最后再加上限定条件i不等于k目的是针对全不为0的特殊情况
class Solution(object):
def moveZeroes(self, nums):
"""
:type nums: List[int]
:rtype: None Do not return anything, modify nums in-place instead.
"""
k = 0 # 0到k的元素都是已排好非零数
for i in range(len(nums)):
if nums[i] != 0:
if i!= k: # 优化全不为0时的情况
nums[k], nums[i] = nums[i], nums[k]
k = k + 1
else:
k = k + 1
return nums