LeetCode每日一题——1089. 复写零

题目

给你一个长度固定的整数数组 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

思路

本题思路很多，这里用的是遍历一遍数组，发现0元素将数组最后一个元素删除，在这个0元素后再添加一个0。

题解

def duplicateZeros(self, arr: List[int]) -> None:
    """
    Do not return anything, modify arr in-place instead.
    """
    index = 0
    while index < len(arr):
        if arr[index] == 0 and index +1 < len(arr):
            arr.pop()
            arr.insert(index+1, 0)
            index += 2
        else:
            index += 1