python3面试题：给一个数组，如果数组中有0，则在0后面追加一个0，整体的数组长度不变

面试题

给一个数组，如果数组中有0，则在0后面追加一个0，整体的数组长度不变，要求不能生成新的数组，只能在当前数组下操作
输入：
arr = [1, 2, 4, 0, 5, 0, 9, 6]
期望输出：
[1,2,4,0,0,5,0,0]

解决思路

使用递归，每次找到列表中的0，在后面添加一个0，并移除最后一项，得到新的arr
依次类推，第一次是从0位置开始，下一次从添加0后的位置继续

def func(index:int, arr:list):
"""
如果数组中有0，则在0后面追加一个0，整体的数组长度不变
:param index: index是a的索引
:param arr: 传入数组
:return: func()函数 index从插入的位置，继续递归
"""
if index >= len(arr)-1:
return arr
for i in range(index, len(arr)):
if arr[i] == 0:
arr.insert(i+1, 0) # 插入0
arr.pop() # 移除最后一个
return func(i+2, arr) # 递归

if __name__ == '__main__':
arr = [1, 2, 4, 0, 5, 0, 9, 6]
print(func(0, arr))

运行结果：[1, 2, 4, 0, 0, 5, 0, 0]