【LeetCode】70. 爬楼梯

转:

【LeetCode】70. 爬楼梯

从本题中我们可以学到包含重复子问题,可以采用记忆化的方式,复用计算后的值;并用动态规划的思想,找到动态转移方程,采用循环实现。

题目描述:

题目:假设我们需要爬一个楼梯,这个楼梯一共有 N 阶,可以一步跨越 1 个或者 2 个台阶,那么爬完楼梯一共有多少种方式?

示例:

输入:2
输出:2

有2种方式可以爬楼梯,跳1+1阶,跳2阶

示例:

输入:3
输出:3

有3种方法爬楼梯,跳1+1+1阶,1+2阶,2+1阶;

推理:

1阶楼梯,跳1阶,有1种方法;
2阶楼梯,跳1+1阶,跳2阶,有2种方法;
3阶楼梯,跳1+1+1阶,1+2阶,2+1阶,有3种方法;
4阶楼梯,跳1+1+1+1阶,1+2+1阶,1+1+2阶,2+1+1阶,2+2阶,有5中方法;

如果要跳 n 阶台阶,最后一步动作可以是上1阶,也可以上2阶,可以转化为:

  • 如果选择最后上 1 阶到达,则求出上 n - 1 个台阶有多少种方法
  • 如果选择最后上 2 阶到达,则求出上 n - 2 个台阶有多少种方法

以上4阶楼梯举例,选择最后上 1 阶到达,则为 1 + (1+1+1)阶,1 + (2+1)阶,1 + (1+2)阶,括号中的方法,正好是上 3 阶楼梯的方法;选择最后上 2 阶到达,则为 2 + (1+1)阶,2 + (2)阶,括号中的方法,正好是上 2 阶楼梯的方法。

所以最后上 n 阶楼梯可以得出:

fn(n) = fn(n - 1) + fn(n  - 2)

类似是 斐波那契数列 的形式了,可以用递归进行实现。

递归实现代码:

var climbStairs = function(n) {
  if(n === 1) return 1
  if(n === 2) return 2

  return climbStairs(n - 1) + climbStairs(n - 2)
};

console.log(climbStairs(4));  // 5
console.log(climbStairs(20));  // 10946

可以画个图具象表示:

【LeetCode】70. 爬楼梯

递归算法的时间复杂度怎么计算?用子问题个数乘以解决一个子问题需要的时间。

首先计算子问题个数,即递归树中节点的总数。显然二叉树节点总数为指数级别,所以子问题个数为 O(2^n)

然后计算解决一个子问题的时间,在本算法中,没有循环,只有 f(n - 1) + f(n - 2) 一个加法操作,时间为 O(1)。

所以,这个算法的时间复杂度为二者相乘,即 O(2^n),指数级别,随着 n 越大,复杂度会越来越高。

在以上递归过程中,会有很多重复的计算。例如计算上 5 阶梯的方法,则需要计算上 4 阶梯的方法,和上 3 阶梯的方法;要计算上 4 阶梯的方法,则需要计算上 3 阶梯的方法,和上 2 阶梯的方法。

计算上 3 阶梯的方法在第一次计算后,之后又要重新计算,这样会造成重复计算。

这是一个典型的重叠子问题,怎么让重复计算的结果更高效的利用呢?

可以采用记忆化递归的方式,把已经计算好的结果缓存起来,以备遇到已经计算过的数字,可以直接使用,不再耗时计算。

记忆化递归

记忆化递归代码实现:

const climbStairs = function(n) {

  const cache = {}  // 缓存计算过的值

  const loop = (n) => {
    if(n === 1) return 1
    if(n === 2) return 2

    if(!cache[n]){
      cache[n] = loop(n - 1) + loop(n - 2)
    }

    return cache[n]
  }

  return loop(n)
};

console.log(climbStairs(4));  // 5
console.log(climbStairs(20));  // 10946

从上面看出,定义对象来存储已计算好的结果,key 值为上的阶梯数,value 值为阶梯计算后的方法数, 每个阶梯只需要计算一次,可以达到 O(n) 的时间复杂度。

这种方法是 自顶向下 计算,从一个规模较大的原问题 fn(20) 开始,一步步拆分为越来越小的规模计算,直到最后不能拆分为止的 fn(1)fn(2) 为止,然后逐层返回结果。

还有一种方式是 自底向上 计算,先从问题最小规模的  fn(1)fn(2) 开始,不断的扩大规模,直到推导出最终原问题 fn(20) 的值,便得到最终的结果。

自底向上 属于动态规划的思路,可以使用循环完成。

动态规划

动态规划代码:


const climbStairs = function(n) {

  const result = []
  result[0] = 0;  // 0 阶 占位
  result[1] = 1;
  result[2] = 2;

  for(let i = 3; i <= n; i++){
    result[i] =  result[i - 1] + result[i - 2]
  }


  return result[n]
};

console.log(climbStairs(4));  // 5
console.log(climbStairs(20));  // 10946

在动态规划中有一个 动态转移方程 的概念,实际上就是描述问题结构的数学形式:
**
【LeetCode】70. 爬楼梯

听起来很高深,实际上可以把 fn(n) 当做一个状态,这个状态是由状态 fn(n-1)fn(n-2) 相加转移而来的,通过不断的循环,转态不断的转移到要求值的 n 上,仅此而已。

上面的代码还可以进一步简化,当前的状态只和两个状态相关,记录两个状态便能够得到另一个状态,在循环过程中记录两个状态即可。

简化代码:

const climbStairs = function(n) {

  if(n < 1) return 0
  if(n === 1 ) return 1
  if(n === 2 ) return 2

  let current = 2; // 前一个
  let prev = 1; // 前前一个
  let  sum = 0;
  for(let i = 3; i <= n; i++){
    sum = current + prev
    prev = current
    current = sum
  }
  return current
};
console.log(climbStairs(4));  // 5
console.log(climbStairs(20));  // 10946

总结:

以上是对这道题的解析发现,可以使用斐波那契额的思路,采用递归的方式实现,时间复杂度在 O(2^n),成指数级上升;

又从中看出有重叠子问题,采取记忆化递归,将重复计算的值缓存起来,以免多次计算,时间复杂度降到了 O(n)

后有采用了动态规划的方式,得到转移方程式,采用循环的方式实现。

如果对你有帮助,请关注【前端技能解锁】:

【LeetCode】70. 爬楼梯

转:

【LeetCode】70. 爬楼梯

上一篇:Python——条件概率运行代码


下一篇:移动开发技术!卧薪尝胆70天内推入职阿里,经典好文