一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:
输入:n = 2
输出:2
示例 2:
输入:n = 7
输出:21
示例 3:
输入:n = 0
输出:1
提示:
0 <= n <= 100
解题思路:
- 只有一个台阶,只需要跳一阶,一种
- 两个台阶,可以每次跳一阶,也可以跳两阶,两种
- 三个台阶,可以每次跳一阶;也可以先跳一阶,再跳两阶;也可以先跳两阶,再跳一阶,三种
- 根据上述情况可知,最后一跳可以跳一阶或跳两阶,由此可得出f(n)=f(n-1)+f(n-2)
class Solution {
public:
int numWays(int n) {
const int MOD = 1000000007;
// 动态规划
// vector<int>dp;
// dp.push_back(1);
// dp.push_back(1);
// dp.push_back(2);
// for(int i=3;i<=n;++i){
// dp.push_back((dp[i-1]+dp[i-2])%MOD);
// }
// return dp[n];
// 动态规划优化,使用滚动数组
if(n==0||n==1) {
return 1;
}
int p=1,q=1,r;
for(int i=2; i<=n; i++) {
r=(p+q)%MOD;
p=q;
q=r;
}
return r;
}
};