三步问题。有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶或3阶。实现一种方法,计算小孩有多少种上楼梯的方式。结果可能很大,你需要对结果模1000000007。
示例1:
输入:n = 3
输出:4
说明: 有四种走法
示例2:
输入:n = 5
输出:13
提示:
n范围在[1, 1000000]之间
java代码:
class Solution {
public int waysToStep(int n) {
if (n <= 2) {
return n;
}
long MOD = 1000000007;
long dp1 = 1;
long dp2 = 2;
long dp3 = 4;
for (int i = 4; i <= n; i++) {
long temp = (dp1 + dp2 + dp3) % MOD;
dp1 = dp2;
dp2 = dp3;
dp3 = temp;
}
int res = (int) (dp3 % MOD);
return res;
}
}