移动零
easy
给定一个数组
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
**进阶:**你能尽量减少完成的操作次数吗?
法1:操作次数多,效率非常低
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
for i, num in enumerate(nums):
if num == 0:
nums.remove(0)
nums.append(0)
return nums
法2:双指针(效率高)
演示:
- 步骤一:
- 步骤二
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
i = j = 0#设置双指针
#步骤一
#i遍历数组,j指向为0的下标,当nums[i]不为0时,将nums[i]的值替换到nums[j],然后j向后移动一位
while i < len(nums):
if nums[i] != 0:
nums[j] = nums[i]
j += 1
i += 1
#步骤二
#步骤一已经将所有非0数移到前面去了,现在j在最后一个非0数之后,将j到结尾的数全部赋值为0
while j < len(nums):
nums[j] = 0
j += 1
return nums
我们创建两个指针 i 和 j,第一次遍历的时候指针 j 用来记录当前有多少 非0 元素。即遍历的时候每遇到一个 非0 元素就将其往数组左边挪,第一次遍历完后,j 指针的下标就指向了最后一个 非0 元素下标。
第二次遍历的时候,起始位置就从 j 开始到结束,将剩下的这段区域内的元素全部置为 0。