python 三步问题

三步问题

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

示例1:

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

 输入：n = 5
 输出：13
提示:

n范围在[1, 1000000]之间

提供两种题解

class Solution:
    """
    解题思路一:递归解法
    1.递推公式为 f(n) = f(n-1) + f(n-2) + f(n-3)
    2.添加备忘录 解决递归的重复调用问题
    解题思路二: 非递归解法
    1.类似于斐波那契额数列
    2.假设a, b, c = 1, 2, 4
    3.公式为: d = a + b + c
     """
    
    # 解题思路一
    def __init__(self):
        self.memo = dict()

    def waysToStep(self, n: int) -> int:
        if n == 1 or n == 2:
            return n
        
        if n == 3:
            return 4
        
        if self.memo.get(n):
            return self.memo.get(n)
        
        self.memo[n] = (self.waysToStep(n-1) + self.waysToStep(n-2) + self.waysToStep(n-3)) % 1000000007
        return self.memo[n]
            
    # 解题思路二
    def waysToStep(self, n: int) -> int:
        if n == 1 or n == 2:
            return n
        if n == 3:
            return 4
        # a, b, c 为初始值 d为结果
        a, b, c, d = 1, 2, 4, 0
        for i in range(4, n+1):
            d = (a + b + c) % 1000000007
            a, b, c = b, c, d
        return d