本文参考
出自LeetCode上的题库 —— 移动零
https://leetcode-cn.com/problems/move-zeroes/
移动零问题
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。注意 ,必须在不复制数组的情况下原地对数组进行操作
示例1:
输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]
示例2:
输入:nums = [0]
输出:[0]
解题思路
第一种解法,先遍历一遍原列表,记录0的个数的同时将0从列表中删除,然后再在末尾添加对应个数的0
第二种解法利用双指针,我们定义指针i和指针j:
1)当字符串前部都为非0值时,使两个指针同时向前移动
2)遇到0值后,指针j停留在原处
3)指针i继续向右移动,直到找到一个非0值,此时交换两个指针所指的值
4)交换完成后,指针i和指针j继续向右移动,指针j向右移动一位后一定再次停留在0值的位置上
5)重复3)和4),直到指针i完成列表的遍历
两种解法都占用$O(n)$的时间复杂度,但第二种解法更快,并且仅需$O(1)$的空间复杂度
O(n)空间复杂度解法
class Solution:
# 移除0后再添加
def move_zeroes_1(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
i = 0
cnt = 0
while i != len(nums):
if nums[i] == 0:
cnt += 1
nums.pop(i)
else:
i += 1
for j in range(cnt):
nums.append(0)
O(1)空间复杂度解法(双指针)
def move_zeroes_4(self, nums: List[int]) -> None:
if not nums:
return
# 开始时,若没有遇到 0,则指针 i 和指针 j 一起移动
# 碰到 0 后,指针 j 始终会指向 0
j = 0
for i in range(len(nums)):
# 当前元素 !=0 时进行交换
if nums[i]:
if i != j:
nums[j], nums[i] = nums[i], nums[j]
j += 1