面试题 08.01. 三步问题

题目

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

示例1：

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

示例2：

输入：n = 5
输出：13
说明：有13种走法

分析： 动态规划问题，最后一步的走法有走1阶，2阶，3阶三种可能，因此走到第n阶总共的走法$F(n)=F(n-1)+F(n-2)+F(n-3)$, (n>=3)，我们可以从初始的第1，2，3阶递推到第n阶
定义F(0)=1 满足
$F(3)=F(2)+F(1)+F(0)$
$F(4)=F(3)+F(2)+F(1)$
以此类推，从式中可以看出，规律满足斐波那契数列

方法1： 通过一个数组，存储0，1，2阶三个初始步数对应的值，超过3阶通过对3取余更新对应位置数组的值
对mod取余是防止数值溢出

class Solution:
    def waysToStep(self, n: int) -> int:
        count = [1, 1, 2]
        mod = 1000000007
        if n < 3:
            return count[n]
        for i in range(3, n+1):
            count[i%3] += (count[(i-1)%3] + count[(i-2)%3])%mod
        return count[n%3]%mod

方法2： 斐波那契数列递推，创建一个中间变量值，用来存储递推出的值

class Solution:
    def waysToStep(self, n: int) -> int:
        if n == 1:
            return 1
        elif n == 2:
            return 2
        elif n == 3:
            return 4
        mod = 1000000007
        a = 1
        b = 2
        c = 4
        for i in range(n-3):
            d = a + b + c
            a = b
            b = c
            c = d % mod
        return c