python 爬楼梯

【题目描述】

 假设一段楼梯共n(n>1)个台阶，小朋友一步最多能上3个台阶，那么小朋友上这段楼梯一共有多少种方法。

【源代码程序】

def count_ways(n):

    # 创建一个数组来存储不同台阶数目时的方法数

    ways = [0] * (n + 1)

 

    # 初始条件，0个台阶有1种方法，1个台阶有1种方法

    ways[0] = 1

    ways[1] = 1

    ways[2] = 2

 

    # 计算不同台阶数目时的方法数

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

        ways[i] = ways[i - 1] + ways[i - 2] + ways[i - 3]

 

    return ways[n]

 

# 输入楼梯台阶的数目n

n = int(input())

 

# 输出小朋友上这段楼梯的方法数

print(count_ways(n))

【运行测试】