给一个数组,如果数组中有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]
2021年第八期《python接口web自动化+测试开发》7.17号开学(课程全面升级!) 加量不加价(新增postman, 赠送selenium和python基础2个课)