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 ])
下一篇:
【算法】面试笔试中的贪心算法