Python3 三步爬楼梯问题

Python3 三步爬楼梯问题

原题地址 https://leetcode-cn.com/problems/three-steps-problem-lcci/

题目:

三步问题。有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶或3阶。实现一种方法,计算小孩有多少种上楼梯的方式。结果可能很大,你需要对结果模1000000007。

示例1:

 输入:n = 3 
 输出:4
 说明: 有四种走法

示例2:

 输入:n = 5
 输出:13

解题:

class Solution:
    def waysToStep(self, n: int) -> int:
        if n==0:
            return 0
        if n==1:
            return 1
        if n==2:
            return 2
        if n==3:
            return 4
        # 本题用动态规划的思路解题,但是不需要n个数组空间,只需3个记录
        e1 = 1
        e2 = 2
        e3 = 4
        for i in range(4, n+1):
            e4 = (e1 + e2 + e3)%1000000007 # 注意!取模一定要在这里就算好,而不是放到最后一步
            e1 = e2
            e2 = e3
            e3 = e4
        return e3
发布了24 篇原创文章 · 获赞 0 · 访问量 421

猜你喜欢

转载自blog.csdn.net/qq_18138105/article/details/105162449