题目描述:
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
来源:力扣(LeetCode)
方法1 递归
思路: 递推公式为f(n)=f(n-1)+f(n-2)。利用斐波那契数列递归方法可以实现。
代码:
class Solution {
public int numWays(int n) {
if(n <= 1)
return 1;
if(n == 2)
return 2;
// 按照题目要求取模
return (numWays(n-1) + numWays(n-2)) % 1000000007;
}
}
分析: 太多重复子运算,超时。
结果:
方法2 动态规划
思路:
- 状态定义:dp为一维数组,其中dp[n]的值为n个台阶的跳法。
- 转移方程:dp[n]=dp[n-1]+dp[n-2]。
- 初始状态:dp[0]=1,dp[1]=1。
- 返回值:dp[n]的值。
状态压缩: 由于dp中第i项至于前两项有关,可以使用三个变量,将空间复杂度降为O(1)。
代码:
class Solution {
public int numWays(int n) {
int a = 1, b = 1, sum = 1;
for(int i=2; i<=n; i++){
// 在计算过程中进行取模,防止整型溢出
sum = (a + b) % 1000000007;
a = b;
b = sum;
}
return sum;
}
}
分析: 时间复杂度为O(n),空间复杂度为O(1)。
结果: