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 0if n==1:return 1if n==2:return 2if n==3:return 4# 本题用动态规划的思路解题,但是不需要n个数组空间,只需3个记录e1 = 1e2 = 2e3 = 4for i in range(4, n+1):e4 = (e1 + e2 + e3)%1000000007 # 注意!取模一定要在这里就算好,而不是放到最后一步e1 = e2e2 = e3e3 = e4return e3