LeetCode面试题 08.01. 三步问题(简单,动态规划,python)

1. 题目

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

示例1:

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/three-steps-problem-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题思路

记录 dp1 为走1步到达当前阶梯的走法；同理，dp2为走2步；dp3为走3步。

class Solution:
    def waysToStep(self, n: int) -> int:
        dp1 = 1
        dp2 = 0
        dp3 = 0
        for i in range(1,n+1):
            dp = (dp1 + dp2 + dp3)%(1000000007)
            dp3 = dp2
            dp2 = dp1
            dp1 = dp
        return dp1