力扣第283题-移动0

题目描述

给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。

请注意 ,必须在不复制数组的情况下原地对数组进行操作。

示例 1:

输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]
示例 2:

输入: nums = [0]
输出: [0]
 
提示:

1 <= nums.length <= 104
-231 <= nums[i] <= 231 - 1

解题思路

这道题也是比较简单。思路如下:使用一个指针index指示下次要存入当前元素的位置(从0开始),然后遍历列表,每当遇到一个不为0的数,就将这个数存储到nums列表的index索引,然后index再加一,指向下次要存入数据的位置。最后只需要将index及其以后的位置全都填上0即可。

代码实现:注意看题目要求,不要使用辅助数组、不要返回任何数据,只需要在原列表上进行操作即可。

class Solution:
    def moveZeroes(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        index = 0
        for i in range(len(nums)):
            if nums[i] != 0:
                nums[index] = nums[i]
                index += 1
        for i in range(index, len(nums)):
            nums[i] = 0