题目描述
三步问题。有个小孩正在上楼梯,楼梯有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];
}
};