算法随笔_31:移动零

程序趣谈2025-01-30 16:33

上一篇:算法随笔_30: 去除重复字母-CSDN博客

=====

题目描述如下:

给定一个数组 nums，编写一个函数将所有 0 移动到数组的末尾，同时保持非零元素的相对顺序。

请注意 ，必须在不复制数组的情况下原地对数组进行操作。

示例 1:

复制代码

输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]

=====

算法思路:

题目要求原地对数组进行操作。我们可以设两个下标指针p1，p2，p1=0，p2=0。

然后从左往右枚举数组，先不断移动p2，如果p2处不为0，我们把p1的元素和p2所指元素交换，然后继续移动p1。如果p2处为0，我们继续移动p2。

以此类推，直到p2到达数组末尾。此时的数组已经符合题目要求。

复制代码

class Solution(object):
    def moveZeroes(self, nums):
        
        p1=0
        p2=0
        nums_len=len(nums)
        while p2<nums_len:
            if nums[p2]!=0:
                nums[p1],nums[p2]=nums[p2],nums[p1]
                p1+=1
            
            p2+=1

此算法的时间复杂度为O(n) 。