一、题目描述
给定一个数组 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(暴力解法,遇到长数组会超时)
思路:
1、从头遍历整个数组,遍历到0就从后面的数组中找到最近的一个非0值交换位置
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
for i in range(len(nums)):
if nums[i] == 0 and i != len(nums) - 1:
for j in range(len(nums[i+1:])):
if nums[i+1:][j] != 0:
nums[i] = nums[i+1:][j]
nums[i+j+1] = 0
break
解法1存在如下问题:
1、通过双重循环逐个查找零并替换,这在最坏情况下(如全零数组)会导致O(n²)的时间复杂度【说人话就是数组len值一大就会超时】。
2、切片操作(如nums[i+1:])会创建新列表,虽然不影响正确性,但增加了不必要的计算。
解法2(双指针)
思路:
1、一个指针α遍历整个数组,全力去找所有的非0值
2、另一个指针β先指向数组最前端,用来存放指针α找到的第一个非0值
3、第一个位置满了,指针β指向的位置就加一,用来存放指针α找到的第二个非0值
4、以此类推
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
b = 0
for a in range(len(nums)):
if nums[a] != 0:
temp = nums[a]
nums[a] = 0
nums[b] = temp
b = b + 1
*感慨一下,一个好的思路真的能节省大量时间,为什么我就想不到呢?o(╥﹏╥)o *