假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
示例 1:
输入:n = 2
输出:2
解释:有两种方法可以爬到楼顶。
- 1 阶 + 1 阶
- 2 阶
示例 2:
输入:n = 3
输出:3
解释:有三种方法可以爬到楼顶。
4. 1 阶 + 1 阶 + 1 阶
5. 1 阶 + 2 阶
6. 2 阶 + 1 阶
这道题最简单的方法是利用斐波那契数列来计算
var climbStairs = function(n) {
if (n === 1 || n === 2) {
return n;
}
let number = climbStairs(n - 1) + climbStairs(n - 2);
return number;
}
但是这么做在n为46的时候在LeetCode中会递归超时
所以对动态规划进行优化
用一个数组 dp 存放中间子问题的结果。
定义dp[i]:爬 i 级楼梯的方式数。从base case: dp[0], dp[1]出发,顺序计算,直到算出dp[i]
const climbStairs = (n) => {
const dp = new Array(n + 1).fill(0);
dp[0] = 1;
dp[1] = 1;
for (let i = 2; i < dp.length; i++) {
dp[i] = dp[i - 2] + dp[i - 1];
}
return dp[n];
}
const dp = new Array(n + 1).fill(0);
这句话中,new Array(n + 1)创建了一个数组里面有n+1个数
new Array(n + 1).fill(0);表示里面的数都用0替代,如果不写0则都是undefined
n + 1如果变为n ,n = 2 的时候会报错