https://leetcode-cn.com/problems/climbing-stairs/
原题
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
示例 1:
输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。
1. 1 阶 + 1 阶
2. 2 阶
示例 2:
输入: 3
输出: 3
解释: 有三种方法可以爬到楼顶。
1. 1 阶 + 1 阶 + 1 阶
2. 1 阶 + 2 阶
3. 2 阶 + 1 阶
题目分析
这道题我们可以先分析下,例如爬到 5 阶的楼顶需要多少种方法,其实是求爬到 4 阶有几种方法加爬到 3 阶有几种方法。这里我们需要判断下临界值,如果 n === 1 时,只有 1 种方法,如果 n === 2,则有 2 种方法。再看图中,climbStairs(3) 计算了 3 次,所以我们可以使用数组来保存下爬到 n 阶需要几种方法。
const climbStairs = (n) => {
const cache = []
return dp(n, cache)
}
const dp = (n, cache) => {
if (n <= 0) return 0
if (n === 1) return 1
if (n === 2) return 2
if (cache[n]) return cache[n]
cache[n] = dp(n - 1, cache) + dp(n - 2, cache)
return cache[n]
}
除了递归的方法,我们可以使用从下到上的迭代先计算出来 cache 数组。
const climbStairs = (n) => {
const dp = [0, 1, 2]
for (let i = 3; i <= n + 1; i++) {
dp[i] = dp[i - 1] + dp[i - 2]
}
return dp[n]
}