LeeCode 面试题 08.01

LeeCode 面试题 08.01

题目描述

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

示例

1.输入 n=3，则有4种上楼梯的方式：(1,1,1)(1,2)(3)(2,1)
2.输入n=4，则有7种上楼梯的方式：(3,1)(1,3)(1,2,1)(1,1,2)(2,1,1)(1,1,1,1)(2,2)

解题思路

三步问题，采用动态规划进行计算，对于n层楼梯的走法，可以分为三种情况：1.剩一层楼梯时，一步走一层，2.剩两层楼梯时，一步走两层，3.剩余三层楼梯时，一步走三层
可得公式：
f[n] = f[n-1] + f[n-2] + f[n-3] (n>=3)

又因为一次可以上1、2、3阶楼梯，所以有
f(1) = 1 (1)
f(2) = 2 (1,1)(2)
f(3) = 4 (1,1,1)(1,2)(3)(2,1)

Java代码实现

class Solution {
    public int waysToStep(int n) {
        if(n<=2){
            return n;
        }
        int [] temp = new int[n+1];
        temp[1] = 1;
        temp[2] = 2;
        temp[3] = 4;
        for(int i=4;i<n+1;i++){
            temp[i] = (temp[i-1]+(temp[i-2]+temp[i-3])%1000000007) %1000000007;
        }
        return temp[n];
    }
}