【算法基础】杨辉三角

1.题目描述

输入:5

输入:

1
1 1
1 2 1
1 3 3 1
1 4 6 4 1

2.python代码

def test(n):
    res=[]
    if n==0:
        return None
    elif n==1:
        l1=[n]
        res.append(l1)
        return res
    else:
        _res=test(n-1)  # 获取前面n-1列的res
        res_n=_res[-1]  # 获取n-1列的元素
        l1=[]
        for i in range(len(res_n)+1):
            if i==0 or i==n-1:
                l1.append(1)
            else:
                l1.append(res_n[i]+res_n[i-1])
        _res.append(l1)
        return _res
    return None

3.测试

for i in test(10):
    print(i)

4.效率时间复杂度测试

from timeit import Timer
# 参数1表示函数的调用,参数2表示__main__表示当前模块,指定为当前模块下的test函数
t1=Timer('test(100)','from __main__ import test')  

# number用来指定测试的测试,测试完成之后求取平均值
print(t1.timeit(number=1000),'seconds')