LeetCode 283. 移动零(Python)

给定一个数组 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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章