Leetcode｜70. 爬楼梯【笔记】

链接

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 阶

关键

关键在于斐波那契数列，也就是得发现f(n) = f(n-1) + f(n-2)
但是没想到得出这种关系就能解题
第n层楼梯的方法由n-1层的方法和n-2层的方法组成

• 因为n-1层方法再上1个台阶就是n层，n-2层方法再上2个台阶就也是n层；所以把二者加在一起就是n层的总方法

思路1

# 直接递归解法，容易超时，python可以加个缓存装饰器，这样也算是将递归转换成迭代的形式了
# 除了这种方式，还有增加步长来递归，变相的减少了重复计算
# 还有一种方法，在递归的同时，用数组记忆之前得到的结果，也是减少重复计算
class Solution:
    @functools.lru_cache(100)  # 缓存装饰器
    def climbStairs(self, n: int) -> int:
        if n == 1: return 1
        if n == 2: return 2
        return self.climbStairs(n-1) + self.climbStairs(n-2)

• 缓存装饰器
  functools.lru_cache(maxsize=128, typed=False)
  • maxsize代表缓存的内存占用值，超过这个值之后，就的结果就会被释放，然后将新的计算结果进行缓存,其值应当设为2的幂
  • typed若为True，则会把不同的参数类型得到的结果分开保存
• 递归：程序调用自身

思路2

• 初始状态：dp[1]=1,dp[2]=2
• 状态转化公式：dp[i] = dp[i-1] + dp[i-2]
• 用字典的话会比列表快

# 直接DP，新建一个字典或者数组来存储以前的变量，空间复杂度O(n)
class Solution:
    def climbStairs(self, n: int) -> int:
        dp = {}
        dp[1] = 1
        dp[2] = 2
        for i in range(3,n+1):
            dp[i] = dp[i-1] + dp[i-2]
        return dp[n]

• ｛｝表示字典

思路3

# 还是DP，只不过是只存储前两个元素，减少了空间，空间复杂度O(1)
class Solution:
    def climbStairs(self, n: int) -> int:
        if n==1 or n==2: return n
        a, b, temp = 1, 2, 0
        for i in range(3,n+1):
            temp = a + b
            a = b
            b = temp
        return temp

思路4?

# 直接斐波那契数列的计算公式喽
class Solution:
    def climbStairs(self, n: int) -> int:
        import math
        sqrt5=5**0.5
        fibin=math.pow((1+sqrt5)/2,n+1)-math.pow((1-sqrt5)/2,n+1)
        return int(fibin/sqrt5)

相似题：剑指 Offer 10- II. 青蛙跳台阶问题

• 比leetcode70多了一个初始0的状态：

class Solution:
    def numWays(self, n: int) -> int:
        dp = {}
        dp[0] = 1
        dp[1] = 1
        dp[2] = 2
        for i in range(3, n+1):
            dp[i] = dp[i-1] + dp[i-2]
        return dp[n]%1000000007

扩展：牛客：加到n/上楼梯

• 3种情况：+1，+2，+3

class GoUpstairs:
    def countWays(self, n):
        dp = [0] * (n+1)
        dp[1] = 1
        dp[2] = 2
        dp[3] = 4
        for i in range(4, n+1):
            dp[i] = (dp[i-1] + dp[i-2] + dp[i-3]) % 1000000007
        return dp[n]

相关知识

疑问

• 动态规划问题不太懂
  • 关于动态规划，你该了解这些！
• 递归解法得了解
• 斐波那契数列是啥

拓展

• 完全背包问题

相似题型

参考

[1] 70.重拳出击(Python3)🙃汇集了题解区五大主流方法，要学姿势就要学的全面点
[2] Python中最常见括号()、[]、{}的区别
[3] Python入门 数据结构 dict字典
[4] 「代码随想录」70. 爬楼梯【动态规划经典题目】详解
[5] functools.lru_cache装饰器详解