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后的位置繼續

# 作者-上海悠悠 QQ交流群:717225969
# blog地址 https://www.cnblogs.com/yoyoketang/

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]