题目描述:
给你一个长度固定的整数数组 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,索引加1
class Solution:
def duplicateZeros(self, arr: List[int]) -> None:
"""
Do not return anything, modify arr in-place instead.
"""
i=0
while i<len(arr):
if arr[i]==0:
arr.pop()
arr.insert(i+1,0)
i+=1
i+=1
p.s.评论区答案,不用移动数组的做法
先复制数组arr,用复制的数组tmp修改原数组的值:
如果tmp的值为0时,在arr当前索引值改为0,索引加1,
如果此时索引不越界,将arr当前值改为tmp当前值,索引加1
由于为0是,两个操作都执行,所以相当于复写了0
tmp = [x for x in arr]
o = 0
for x in tmp:
if x == 0:
if o < len(arr):
arr[o] = 0
o += 1
if o < len(arr):
arr[o] = x
o += 1