题目描述
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
0 <= n <= 100
示例
输入:n = 2
输出:2
输入:n = 7
输出:21
输入:n = 7
输出:21
解法
经推导归纳得到,此题为斐波那契数列的变形,i.e.将斐波那契数列的前两项由【0, 1】改为【1, 1】
递归
C++
class Solution {
unordered_map<int, int> res;
public:
Solution() {
res[0] = 1;
res[1] = 1;
}
int numWays(int n) {
if (res.count(n)) {
return res[n];
}
return res[n] = (numWays(n - 1) + numWays(n - 2)) % 1000000007;
}
};
动态规划
C++
class Solution {
public:
int numWays(int n) {
if (n < 2) {
return 1;
}
int a = 1;
int b = 1;
int c;
for (int i = 2; i <= n; ++i) {
c = (a + b) % 1000000007;
a = b;
b = c;
}
return c;
}
};