Распаковка вложенных списков

Кодинг-марафон. Задача 8.

Напишите функцию flatten, которая будет принимать на вход список с любой вложенностью (список в списке), а также второй параметр — глубину распаковки.

Функция должна вернуть список, где вложенные элементы соответствующего уровня распакованы, то есть вложенное превращается в плоское.

Если второй аргумент не передан, функция должна вернуть одномерный список, то есть все вложенные списки должны быть распакованы.

Сигнатура: def flatten(a_list:list, depth:int=0)->list

Примечания

• Функция не должна бросать исключений!
• Предполагается, что глубина распаковки всегда положительна (если указана).
• Если параметр глубины больше реальной вложенности, то функция просто распаковывает все списки, как если бы глубина не была передана.
• Предполагается, что могут быть только вложенные списки (не кортежи или другие контейнеры).

Примеры

flatten([1, [2, 3]]) == [1, 2, 3]
flatten([1, [2, [3]]], depth=1) == [1, 2, [3]]
flatten([1, [2, [3]]], depth=2) == [1, 2, 3]
flatten([1, [2, 3]], depth=100) == [1, 2, 3]

Варианты решения

def flatten(a_list: list, depth: int = 0) -> list:
    count = 0
    while True:
        result = []
        has_list = False
        for element in a_list:
            if type(element) is list:
                result.extend(element)
                has_list = True
            else:
                result.append(element)
        a_list = result
        count += 1
        if not has_list or (0 < depth == count):
            break
    return result

def flatten(a_list: list, depth: int = -1) -> list:
    result = []
    for element in a_list:
        if depth == 0 or type(element) is not list:
            result.append(element)
        else:
            result.extend(flatten(element, depth - 1))
    return result