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