剑指offer 10-II 青蛙跳台阶
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:
输入:n = 2
输出:2
示例 2:
输入:n = 7
输出:21
提示:
0 <= n <= 100
动态规划解析:
状态定义: 设 dp为一维数组,其中 dp[i] 的值代表 斐波那契数列第 $i个数字 。
转移方程: dp[i + 1] = dp[i] + dp[i - 1] ,即对应数列定义 f(n + 1) = f(n) + f(n - 1) ;
初始状态: dp[0] = 1, dp[1] = 1 ,即初始化前两个数字;
返回值:dp[n] ,即斐波那契数列的第 n 个数字。
class Solution {
public int numWays(int n) {
int[] dp=new int[n+1];
if(n==0) return 1;
dp[0]=1;
dp[1]=1;
for(int i=2;i<=n;i++){
dp[i]=dp[i-1]+dp[i-2];
dp[i]%=1000000007;
}
return dp[n];
}
}
空间复杂度优化
由于dp列表第i项只与第i- 1和第i- 2项有关,因此只需要初始化3 个整形变量sum. a. b,利用辅助变量sum使a, b两数字交替前进即可(具体实现见代码人 。
因为节省了dp列表空间,因此空间复杂度降至0(1)。
class Solution {
public int numWays(int n) {
int a = 1, b = 1, sum;
for(int i = 0; i
< n; i++){
sum = (a + b) % 1000000007;
a = b;
b = sum; //上一次的计算结果是下一次计算的参数
}
return a;
}
}
/上一次的计算结果是下一次计算的参数
}
return a;
}
}