题目
三步问题。有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶或3阶。实现一种方法,计算小孩有多少种上楼梯的方式。结果可能很大,你需要对结果模1000000007。
示例
输入:n = 3
输出:4
说明: 有四种走法
输入:n = 5
输出:13
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/three-steps-problem-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法1:动态规划
注意溢出,dp数组取 long型
Java实现
class Solution {
public int waysToStep(int n) {
//下标从0开始,所以申请n+1个空间
if (n == 1) return 1;
if (n == 2) return 2;
if (n == 3) return 4;
long[] dp = new long[n + 1];
dp[0] = 0;
dp[1]= 1;
dp[2] = 2;
dp[3] = 4;
for (int i = 4; i <= n; i++) {
dp[i] = (dp[i - 1] + dp[i - 2] + dp[i - 3]) % 1000000007;
}
return (int)(dp[n] % 1000000007);
}
}