python-leetcode283移动零,一行代码解决

题目

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

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

说明: 必须在原数组上操作,不能拷贝额外的数组。 尽量减少操作次数。

python解法

双指针解法

使用双指针,左指针指向当前已经处理好的序列的尾部,右指针指向待处理序列的头部。

右指针不断向右移动,每次右指针指向非零数,则将左右指针对应的数交换,同时左指针右移。

注意到以下性质:

左指针左边均为非零数;

右指针左边直到左指针处均为零。

因此每次交换,都是将左指针的零与右指针的非零数交换,且非零数的相对顺序并未改变。

class Solution:
    def moveZeroes(self, nums: List[int]) -> None:
        l = r = 0  # l代表左边,r代表右边
        while r < len(nums):
            if nums[r] != 0:
                nums[l], nums[r] = nums[r], nums[l]
                l += 1
            r += 1
一行代码
class Solution:
    def moveZeroes(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        
        nums.sort(key=lambda x:x==0)

解释

其中利用了sort自定义比较函数(利用 lambda)排序的三种方式:1.按照component 2.按照key 3.按照reverse 例如:

students  =  [( john ,  A ,  15 ), ( jane ,  B ,  12 ), ( dave ,  B ,  10 )]
sorted (students, key = lambda  student : student[ 2 ])
经验分享 程序员 微信小程序 职场和发展