力扣--面试题 08.01.三步问题

题目描述

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

示例1:

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

输入：n = 5
输出：13

代码示例

class Solution {
public:
    int waysToStep(int n) {
        if(n <= 2) return n;
        if(n == 3) return 4;
        int d[n+1];

        d[1] = 1;
        d[2] = 2;
        d[3] = 4;
        for(int i=4; i<=n; i++){
            d[i] = (d[i-1] + d[i-2]) % 1000000007 +d[i-3];
            d[i] %= 1000000007;
        }
        return d[n];
    }
};