问题
来自力扣:
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:
输入:n = 2
输出:2
示例 2:
输入:n = 7
输出:21
示例 3:
输入:n = 0
输出:1
提示:
0 <= n <= 100
代码
class Solution {
public:
int numWays(int n) {
if (n==0||n==1) return 1;
int Sn_2 =1,Sn_1=1;
for (int i=0;i<n-1 ;i++){
Sn_1 = Sn_1 + Sn_2;
Sn_2 = Sn_1 - Sn_2;
Sn_1 %= int(1e9+7);
}
return Sn_1;
}
};
这个代码和裴波那契数列的那个大致一样。
思路:如果一个青蛙要跳上第n个台阶,那么,他最终有两种情况:1从第n-2阶跳2级;2从第n-1阶跳1级。
方法:只需要求出跳上第n-2和跳上第n-1阶各自的跳法,然后相加即可。
示例代码
class Solution {
public:
#define N 110
const int mod = 1e9 + 7;
int dp[N];
int numWays(int n) {
if (n <= 1) return 1;
dp[0] = 1;
dp[1] = 1;
for (int i = 2; i <= n; i ++)
dp[i] = (dp[i - 1] + dp[i - 2]) % mod;
return dp[n];
}
};
示例代码用了数组,我感觉是没有必要的,毕竟我并不需要使用旧的数据。