【python算法书】动态规划算法，爬楼梯问题？

题目：窝窝家住在二楼，每次回家都需要经过一个有10层台阶的楼梯。窝窝每次可以选择一步走一级台阶或者一步都两级台阶。请帮窝窝计算他从楼下到家一共有多少种走法。

分析：

code：

def upstairs(n):

    a = 1  # 初始化边界值

    b = 2

    temp = 0

    if n < 1:

        print(0)

    if n == 1:

        print(1)

    if n == 2:

        print(2)

    if n >= 3:

        for i in range(3, n + 1):

            temp = a + b

            a = b

            b = temp

        print(temp)

if __name__ == "__main__":

    upstairs(3)