三步问题。有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶或3阶。实现一种方法,计算小孩有多少种上楼梯的方式。结果可能很大,你需要对结果模1000000007。
示例1:
输入:n = 3
输出:4
说明: 有四种走法
示例2:
输入:n = 5
输出:13
提示:
n范围在[1, 1000000]之间
解题思路(动态规划):
要想走上当前第n阶台阶,可以选择从第n-1、第n-2和第n-3阶台阶上楼梯,所以上楼梯的方式是上述累加和
代码实现:
class Solution {
public:
int waysToStep(int n) {
if(n<=2){
return n;
}
if(n==3){
return 4;
}
int* dp=new int[n+1];
dp[0]=0;
dp[1]=1;
dp[2]=2;
dp[3]=4;
int mod=1000000007;
for(int i=4;i<=n;++i){
dp[i]=((dp[i-1]%mod+dp[i-2]%mod)%mod+dp[i-3]%mod)%mod;
}
return dp[n];
}
};