三步问题。有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶或3阶。实现一种方法,计算小孩有多少种上楼梯的方式。结果可能很大,你需要对结果模1000000007。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/three-steps-problem-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
int waysToStep(int n) {
if(n<=2){
return n;
}
if(n==3){
return 4;
}
vector<int>a(n+1,0);
a[1]=1;
a[2]=2;
a[3]=4;
for(int i=4;i<=n;i++){
a[i]=(a[i-1]+a[i-2])%1000000007+a[i-3];
a[i]%=1000000007;
}
return a[n];
}
};