假设你正在爬楼梯。需要 n 阶你才能到达楼顶。 每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢? 注意:给定 n是一个正整数。
动态规划
思路:
两种方式:
- 1阶或者2阶,假设第一次是1阶,那么剩下的n-1阶台阶,就有f(n-1)种爬法。
- 假设第一次是2阶,那么剩下的n-2阶台阶,就有f(n-2)种爬法。
- 所以总方法数为f(n) = f(n-1) + f(n-2)
以上是动态规划的转移方程,下面我们来讨论边界条件。
- 只有一阶楼梯的时候只有一种方案,即f(1) = 1。
- 只有两阶的时候有两种方案,爬一个或者两个台阶,即f(2) = 2。
- 这两个作为边界条件就可以继续向后推导出第 n 级的正确结果。我们不妨写几项来验证一下,根据转移方程得到 f(3) = 3,f(4)=5…我们把这些情况都枚举出来,发现计算的结果是正确的。
我们不难通过转移方程和边界条件给出一个时间复杂度和空间复杂度都是 O(n) 的实现,但是由于这里的 f(x)只和 f(x - 1)与 f(x−2) 有关,所以我们可以用「滚动数组思想」把空间复杂度优化成 O(1)。
/**
* @param {number} n
* @return {number}
*/
var climbStairs = function(n) {
let f = 1,g=2;
while(--n){
g += f;
f = g - f;
}
return f;
};