谁还不得学学数据结构 leet code.70(动态规划)

leet code.70
爬梯子假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢？
注意：给定 n 是一个正整数。

def climbStairs(n):
    mem=[]
    #mem[0]=1
    #mem[1]=2
    mem=[1,2]
    ###n对应到数组中是n-1
    for i in range(2,n):
        #mem[i]=mem[i-1]+mem[i-2]
        mem.append(mem[i-1]+mem[i-2])
    return mem[-1]
if __name__ == '__main__':
    a=climbStairs(2)
    print(a)