python每日一题——4移动0

WY_记录2023-11-25 10:41

题目

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

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

示例 1:

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

输出: [1,3,12,0,0]

示例 2:

输入: nums = [0]

输出: [0]

答案

这道题可以使用双指针的方法解决，一个指针用来遍历数组，一个指针用来记录0的末尾位置。

具体步骤如下：

定义两个指针，一个指向当前遍历的位置，一个指向0的末尾位置。

遍历数组，如果当前元素为0，则将其移动到0的末尾位置，并将0的末尾位置指针向前移动一位。

如果当前元素不为0，则将其移动到当前位置，并将当前位置指针向前移动一位。

重复上述步骤，直到遍历完整个数组。

代码如下：

python 复制代码

def moveZeroes(nums):  
    if not nums:  
        return []  
    left = 0  # 当前遍历位置指针  
    right = 0  # 0的末尾位置指针  
    n = len(nums)  
    while left < n:  
        if nums[left] == 0:  
            nums[left], nums[right] = nums[right], nums[left]  
            right += 1  
        else:  
            nums[left] = nums[left] + 1  # 非零元素加上一个非零值，保证相对顺序不变  
            left += 1  
    return nums

时间复杂度为O(n)，空间复杂度为O(1)。