代码随想录算法训练营第三十二天|动态规划理论基础、LeetCode 509. 斐波那契数、70. 爬楼梯、746. 使用最小花费爬楼梯

发布于:2025-07-14 ⋅ 阅读:(12) ⋅ 点赞:(0)

目录

LeetCode 509. 斐波那契数

70. 爬楼梯

746. 使用最小花费爬楼梯

感想


文档讲解:代码随想录

动态规划,英文:Dynamic Programming,简称DP,如果某一问题有很多重叠子问题,使用动态规划是最有效的。

所以动态规划每一个状态一定是由上一个状态推导出来的这一点就区别于贪心,贪心没有状态推导,而是从局部直接选最优的。

解题步骤:

  1. 确定dp数组(dp table)以及下标的含义

  2. 确定递推公式

  3. dp数组如何初始化

  4. 确定遍历顺序

  5. 举例推导dp数组

为什么要先确定递推公式,然后在考虑初始化呢?因为一些情况是递推公式决定了dp数组要如何初始化

做动规的题目,写代码之前一定要把状态转移在dp数组上的具体情况模拟一遍,心中有数,确定最后推出的是想要的结果。然后再写代码,如果代码没通过就打印dp数组,看看是不是和自己预先推导的哪里不一样。

题目类型有基础题目、背包问题、打家劫舍、股票问题、子序列问题 五大类

LeetCode 509. 斐波那契数

文档讲解:代码随想录

视频讲解:手把手带你入门动态规划 | LeetCode:509.斐波那契数_哔哩哔哩_bilibili

状态:简单题,做对了,要注意特殊情况的处理,怎么返回

class Solution:
    def fib(self, n: int) -> int:
        if n == 0: # 边界条件
            return 0
        if n == 1:
            return 1
        dp = [0]*(n+1) # step1:dp数组 dp[i]是第i个斐波那契数的值
        dp[1] = 1 # step3:dp数组初始化
        for i in range(2, n+1): # step4:因为要用到前面的状态,所以遍历顺序从前向后
            dp[i] = dp[i-1] + dp[i-2] # step2:递推公式/状态转移方程
        return dp[n]

时间复杂度O(n)

空间复杂度O(n)

当前状态dp[i]只依赖前面两个状态dp[i-1]和dp[i-2],所以可以不用维护一整个数组,而是3个量就可以了,得到上面程序的精简版:

class Solution:
    def fib(self, n: int) -> int:
        if n <= 1:
            return n
        dp = [0]*2
        dp[1] = 1
        for i in range(2, n+1):
            sum = dp[0] + dp[1]
            dp[0] = dp[1]
            dp[1] = sum
        return dp[1]

 时间复杂度O(n)

空间复杂度O(1)

70. 爬楼梯

文档讲解:代码随想录

视频讲解:

状态:去年7月11日做过,一周年了,结果还是不会做。 想dp数组的定义,想了好久,不确定到底该怎么定义,到底是第i走了几个台阶 or 一共走了多少台阶 or 剩下多少台阶,但最后需要返回的是方法数量,这该咋办呢?

我的错误思路:

        # dp[i]是走完第i个台阶后,一共走了多少台阶
        # 递推公式 dp[i] = dp[i-1] + 1或2
        dp = [0] * n # 每次都走一个台阶,最多n步
        dp[0] = 1  # 但也有可能是2?
        if dp[-1] < n:  # 该确定遍历顺序了,怎么遍历?
            dp

正确思路是直接考虑每层楼梯的方法数目:

爬到第一层楼梯有一种方法,爬到二层楼梯有两种方法。

那么第一层楼梯再跨两步就到第三层 ,第二层楼梯再跨一步就到第三层。

所以到第三层楼梯的状态可以由第二层楼梯 和 到第一层楼梯状态推导出来。

动规五部曲:

1. 一维数组 dp[i]: 爬到第i层楼梯,有dp[i]种方法

2. 递推公式:考虑最后一步走几个台阶,有两种情况1或者2。上i-1层楼梯,有dp[i - 1]种方法,那么再跳一个台阶就是dp[i];上i-2层楼梯,有dp[i - 2]种方法,那么再一步跳两个台阶就是dp[i]。 所以爬到第i层楼梯,有dp[i] = dp[i - 1] + dp[i - 2] 种方法。

3. 初始化:从dp数组定义的角度出发,不用考虑dp[0]的初始化,直接dp[1] = 1,dp[2] = 2,然后从i = 3开始递推

4. 从前向后遍历

5. 举例推导dp数组    1  2  3  5  8  13 

斐波那契数列:0、1、1、2、3、5、8、13、21、34……

与斐波那契数列后半截一样的

class Solution:
    def climbStairs(self, n: int) -> int:
        if n == 1:
            return n
        dp = [0]*(n+1) # 因为python下标从0开始,到n 长度为n+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]

注意!一定要注意边界条件的处理!如果不写 if n==1: return n 的话,当n = 1时,dp[2] = 2 处就会报错list out of range。

空间和时间复杂度:O(n)  此题也可以像上一题一样优化空间复杂度为O(1)

拓展:这道题目还可以继续深化,就是一步一个台阶,两个台阶,三个台阶,直到 m个台阶,有多少种方法爬到n阶楼顶。这其实是一个完全背包问题,后面会讲。

746. 使用最小花费爬楼梯

文档讲解:代码随想录

视频讲解:

状态:自己做对了!类比上一题。

动规五部曲:

定义:dp[i]是到达第i个台阶需要支付的最少费用

公式:dp[i] = min((dp[i-1] + cost[i-1]), dp[i-2] + cost[i-2])

初始化:dp[0] = 0   dp[1] = 0       题目中说 “你可以选择从下标为 0 或下标为 1 的台阶开始爬楼梯” 也就是相当于 跳到 下标 0 或者 下标 1 是不花费体力的

顺序:从前向后

举例:

cost = [1,100,1,1,1,100,1,1,100,1]

下标          0    1    2    3    4    5    6    7    8    9   楼顶

dp数组      0     0   1    2   2    3    3    4     4    5    6     确定dp数组长度为len(cost)+1

class Solution:
    def minCostClimbingStairs(self, cost: List[int]) -> int:
        dp = [0]*(len(cost)+1)
        dp[0] = 0
        dp[1] = 0
        for i in range(2,len(cost)+1): # 从2开始推导
            dp[i] = min((dp[i-1] + cost[i-1]), dp[i-2] + cost[i-2])
        return dp[-1]

感想

学习用时:晚上2h + 晚上2h


网站公告

今日签到

点亮在社区的每一天
去签到