LeetCode-Python-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. 1 <= arr.length <= 10000
2. 0 <= arr[i] <= 9

思路：

线性扫描，遇到一个0就把它后面的数依次往后挪一位，然后加一个0到这个0后面。

class Solution(object):
    def duplicateZeros(self, arr):
        """
        :type arr: List[int]
        :rtype: 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