题目描述:
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
链接:青蛙跳台阶问题
1. dp
思路:对于每一次跳的阶数,要么是1层,要么是2层,则到达第n层的次数为dp[n] = dp[n - 1] + dp[n - 2],和斐波那契数列一样,只不过初始条件不一样。
代码:
class Solution {
public:
int numWays(int n) {
if(n == 0)
return 1;
vector<int> dp(n + 1);
dp[0] = dp[1] = 1;//初始条件
for(int i = 2; i <= n; ++i)
dp[i] = (dp[i - 1] + dp[i - 2]) % 1000000007;
return dp[n];
}
}
2. dfs + 回溯
思路:每一次走1层或者2层,对每一次的结果进行保存,然后搜索,等于目标层数时,则计数增加一个。
递归太多,会超时。
代码:
class Solution {
public:
int count = 0;
int sum = 0;
int numWays(int n) {
if(n == 0)
return 1;
if(n == 1)
return 1;
if(n == 2)
return 2;
_numWays(n);
return count;
}
private:
void _numWays(int n)
{
if(sum == n)
{
count = (count + 1) % 1000000007;
return;
}
for(int i = 1; i <= 2; ++i)
{
if(sum + i > n)
return;
sum += i;
_numWays(n);
sum -= i;
}
}
}