编程语言
首页 > 编程语言> > 打败算法 —— 移动零

打败算法 —— 移动零

作者:互联网

本文参考

出自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

标签:nums,复杂度,move,算法,打败,移动,解法,指针
来源: https://www.cnblogs.com/kuluo/p/15959468.html