剑指 Offer 10- II. 青蛙跳台阶问题
题目描述
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 3:
输入:n = 0 输出:1
示例 1:
输入:n = 2 输出:1
示例 2:
输入:n = 5 输出:5
动态规划
动态规划解析:
- 状态定义: dp[i] 代表 斐波那契数列第 i 个数字 。
- 转移方程: dp[i +1] = dp[i] + dp[i - 1] ,即对应数列定义 f(n + 1) = f(n) + f(n - 1);
- 初始状态: dp[0] = 0, dp[1] = 1 ,即初始化前两个数字;
- 返回值: dp[n] ,即斐波那契数列的第 n 个数字
var numWays = function(n) {
const dp = []
dp[0] = 1
dp[1] = 1
for(let i = 2; i <= n; i++){
dp[i] = (dp[i-1] + dp[i-2]) % 1000000007
}
return dp[n]
};
简化为 O(1)空间
var numWays = function(n) {
const dp = [1,1]
for(let i = 2; i <= n; i++){
let sum = (dp[0] + dp[1]) % 1000000007
dp[0] = dp[1]
dp[1] = sum
}
return dp[1]
};