leetCode第283题 移动零
给定一个数组 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
进阶:你能尽量减少完成的操作次数吗?
python对列表的操作十分的便利,不涉及算法的话,只要遍历列表,碰到零元素就删掉,顺便在列表末尾加一个0,代码如下,但效果非常低
## python3
class Solution:
def moveZeroes(self, nums: [int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
for i in range(len(nums)):
if nums[i] == 0:
nums.remove(0)
nums.append(0)
可以引入双指针解决
将指针i,j先指向第一个元素,让i不断后移,如果i指向的是一个非零元素,那么将i指向元素的值直接填入j指向的位置,然后j指针后移;如果i指向的是一个零元素,那么指针i就直接后移,而j不变。
当i已经全部遍历完后,从j的位置开始,直接往后全部填入零。
指针i.j都把列表遍历了一遍,所以时间复杂度应该为O(2n),即线性的时间复杂度O(n)
整个过程只定义了两个指针,空间复杂度应该为O(1)
在时间上得到了很大的提升,效果如下
## python3
class Solution:
def moveZeroes(self, nums: [int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
if len(nums) == 0:
return
j = 0
for i in range(len(nums)):
if nums[i] != 0:
nums[j] = nums[i]
j += 1
for i in range(j,len(nums)):
nums[i] = 0