构造特定数组的逆序拼接 | 豆包MarsCode AI刷题

33 阅读1分钟
def solution(n: int) -> list:
    result = []
    for i in range(1, n + 1):
        # 生成从 n 到 i 的逆序列表
        reverse_list = list(range(n, i - 1, -1))
        # 将这个逆序列表追加到 result 中
        result.extend(reverse_list)
    return result

if __name__ == '__main__':
    print(solution(3) == [3, 2, 1, 3, 2, 3])
    print(solution(4) == [4, 3, 2, 1, 4, 3, 2, 4, 3, 4])
    print(solution(5) == [5, 4, 3, 2, 1, 5, 4, 3, 2, 5, 4, 3, 5, 4, 5])