leetcode_1089 复写零

115 阅读1分钟

要求

给你一个长度固定的整数数组 arr,请你将该数组中出现的每个零都复写一遍,并将其余的元素向右平移。

注意:请不要在超过该数组长度的位置写入元素。

要求:请对输入的数组 就地 进行上述修改,不要从函数返回任何东西。

示例 1:

输入:[1,0,2,3,0,4,5,0]
输出:null
解释:调用函数后,输入的数组将被修改为:[1,0,0,2,3,0,0,4]

示例 2:

输入:[1,2,3]
输出:null
解释:调用函数后,输入的数组将被修改为:[1,2,3]

提示:

1 <= arr.length <= 10000
0 <= arr[i] <= 9

核心代码

class Solution:
    def duplicateZeros(self, arr: List[int]) -> None:
        """
        Do not return anything, modify arr in-place instead.
        """
        l = len(arr)
        i= 0
        while i < l:
            if arr[i] == 0:
                for j in range(l-1,i+1,-1):
                    arr[j] = arr[j-1]
                if i + 1 < l:
                    arr[i + 1] = 0
                i += 2
            else:
                i += 1
        return arr 

image.png

解题思路:使用线性扫描的方式,当我们遇到0的时候,我们将后面的所有元素向后进行平移。