面试题 08.01. 三步问题

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

示例1:

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

示例2:

输入：n = 5
输出：13

提示:

• n范围在[1, 1000000]之间

Python 解答：

class Solution:
    def waysToStep(self, n: int) -> int:
        f1, f2, f3 = 1, 2, 4
        if n == 1:
            return f1
        elif n == 2:
            return f2
        elif n == 3:
            return f3
        while n > 3:
            f1, f2, f3 = f2, f3, (f1+f2+f3) % 1000000007
            n -= 1
        return f3