代码随想录算法训练营第十九天 | 动态规划系列5,6,7,8

动态规划系列5,6,7,8

377 组合总和 Ⅳ

未看解答自己编写的青春版

这道题简直和上一个系列的最后一题如出一撤。

如果求组合数就是外层for循环遍历物品,内层for遍历背包。

如果求排列数就是外层for遍历背包,内层for循环遍历物品。

python 复制代码
class Solution:
    def combinationSum4(self, nums: List[int], target: int) -> int:
        # 初始化dp数组
        dp = [0]*(target+1)
        # dp数组初始化赋值
        dp[0] = 1

        # 开始循环,因为所求为排列问题,最关键的一句话是
        # 顺序不同的序列被视作不同的组合
        # 所以遍历顺序是先遍历背包,再遍历物品
		
		# 下面i这里,第一次写的时候,忘记从1开始遍历, 是从0开始的
		# 不过我们已经初始化了,所以不影响结果
        for i in range(1,target+1):
            for j in range(len(nums)):
                if i >= nums[j]:
                    dp[i] += dp[i-nums[j]]
        return dp[target]

重点

一点感悟:在做本题时,有一点错觉,dp数组的含义是没有异议的,但是因为题目说,不同顺序的排列算作不同的组合,那么加入target=4,当前遍历物品重量为1,则dp[4] += dp[4-1],假如dp[3]中的方式为[3],那么1加入其中,应该有[1,3] [3,1]两种顺序啊,不应该是dp[3]再加某个值吗?

答:不是的!这种情况会造成重复,在我们的代码编写中,物品是按照顺序遍历的,上述dp[4] += dp[4-1],得到的排列就是[3,1] , 就是这一种情况,而[1,3]的情况,则在我们遍历到物品3时,由dp[4] += dp[4-3]得到!

本题和上一篇博客的最后一题,可谓是相辅相成了,这里放上本篇的链接,值得细读。

组合总和 Ⅳ

拓展:我们在上一篇博客,写到了一个爬楼梯问题,如果问题改成,一次可以爬最多M个台阶,求能到达楼顶的方法的个数。

和本题是一模一样的!也是求排列数。

代码随想录的代码

python 复制代码
class Solution:
    def combinationSum4(self, nums: List[int], target: int) -> int:
        dp = [0] * (target + 1)  # 创建动态规划数组,用于存储组合总数
        dp[0] = 1  # 初始化背包容量为0时的组合总数为1

        for i in range(1, target + 1):  # 遍历背包容量
            for j in nums:  # 遍历物品列表
                if i >= j:  # 当背包容量大于等于当前物品重量时
                    dp[i] += dp[i - j]  # 更新组合总数

        return dp[-1]  # 返回背包容量为target时的组合总数

我的代码(当天晚上理解后自己编写)

过。

求排列数的题,用二维DP过不了?自己捋逻辑的话,也是可以觉得有漏洞,但是怎么修改,一下子还没思路,包括后面的"139. 单词拆分"也是一样的情况。

运行错误的二维DP代码

python 复制代码
class Solution:
    def combinationSum4(self, nums: List[int], target: int) -> int:
        n = len(nums)
        dp = [[0]*(target + 1) for _ in range(n+1)]

        dp[0][0] = 1
        for i in range(n+1):
            dp[i][0] = 1
        '''
        for j in range(target+1):
            dp[0][j] = 1
        '''
       
        for j in range(1, target+1):
            for i in range(1,n+1) :
                dp[i][j] = dp[i-1][j]
                if j >= nums[i-1] : 
                    dp[i][j] += dp[i][j-nums[i-1]] 

        return dp[-1][-1]

爬楼梯进阶

放上代码随想录的解答链接。
爬楼梯进阶

能想到爬楼梯问题,问一共有多少种方法,可以看做,完全背包+求排列数。

322 零钱兑换

未看解答自己编写的青春版

写出来了,完全按照动归五部曲做的,写完这道题体会到动归五部曲真好用。

python 复制代码
class Solution:
    def coinChange(self, coins: List[int], amount: int) -> int:
        dp = [inf]*(amount+1)
        dp[0] = 0
        for i in range(1,amount+1):
            for j in range(len(coins)):
                if i >= coins[j]:
                    dp[i] = min(dp[i],dp[i-coins[j]]+1)

        if dp[amount]==inf :
            return -1
        else :
            return dp[amount]

写完这道题后的感受

做动态规划的题目,有时候不需要先把整道题的思路全部想通,一步一步来就好,就按照动归五部曲,这道题一开始我也想不通,不知道为什么背包问题求的是最大,这里求最小怎么弄。

不用管,直接上动归五部曲,求最大,在递推公式那里就是max运算符,求最小,在递推公式那里,就是min运算符。

第一步:确定dp数组的含义,题目求什么就定义为什么,也别想太多,dp[i]定义为:amount为 i 时,所需要的硬币的最小个数。

第二步:确定递推公式,这里按照dp数组的定义,很容易想到,dp[i] = min( dp[i] , dp[i-coins[j]] )

第三步:初始化。这里很重要,也是很容易错的地方。初始化的值和前两步息息相关。一开始我这里就定义错了。 dp[0] = 0 , 因为此时没有硬币组合可以是0,那么 dp[i] 呢 ? 因为递推公式是求min,为了避免初始化对操作符的影响,这里应该初始化为最大值!一开始我想的是初始化为-1,但是这样的话,最后的数组依旧全是-1

第四步:确定遍历顺序。本题求钱币最小个数,那么钱币有顺序和没有顺序都可以,都不影响钱币的最小个数。所以本题并不强调集合是组合还是排列。如果求组合数就是外层for循环遍历物品,内层for遍历背包。如果求排列数就是外层for遍历背包,内层for循环遍历物品。因为{ 5 5 1} 和 { 5 1 5}的钱币个数都是3。我这里的遍历顺序是先遍历的背包,不过感觉也许先遍历物品,更符合习惯。完全背包问题,背包为正序遍历。

第五步:模拟,打印dp数组。

重点

代码随想录的代码

先遍历物品 后遍历背包

python 复制代码
class Solution:
    def coinChange(self, coins: List[int], amount: int) -> int:
        dp = [float('inf')] * (amount + 1)  # 创建动态规划数组,初始值为正无穷大
        dp[0] = 0  # 初始化背包容量为0时的最小硬币数量为0

        for coin in coins:  # 遍历硬币列表,相当于遍历物品
            for i in range(coin, amount + 1):  # 遍历背包容量
                if dp[i - coin] != float('inf'):  # 如果dp[i - coin]不是初始值,则进行状态转移
                    dp[i] = min(dp[i - coin] + 1, dp[i])  # 更新最小硬币数量

        if dp[amount] == float('inf'):  # 如果最终背包容量的最小硬币数量仍为正无穷大,表示无解
            return -1
        return dp[amount]  # 返回背包容量为amount时的最小硬币数量

先遍历物品 后遍历背包(优化版)

这个剪枝操作还可以。

python 复制代码
class Solution:
    def coinChange(self, coins: List[int], amount: int) -> int:
        dp = [float('inf')] * (amount + 1)
        dp[0] = 0

        for coin in coins:
            for i in range(coin, amount + 1): # 进行优化,从能装得下的背包开始计算,则不需要进行比较
                # 更新凑成金额 i 所需的最少硬币数量
                dp[i] = min(dp[i], dp[i - coin] + 1)

        return dp[amount] if dp[amount] != float('inf') else -1

动态规划,也要时刻想着剪枝操作。

我的代码(当天晚上理解后自己编写)

本题首先明确,是完全背包问题,不要求遍历顺序,那么一维DP数组的代码就很好写了。

python 复制代码
class Solution:
    def coinChange(self, coins: List[int], amount: int) -> int:
        dp = [inf]*(amount+1)
        dp[0] = 0
        for i in range(len(coins)):
            for j in range(coins[i],amount+1):      
                dp[j] = min(dp[j],dp[j-coins[i]]+1)

        if dp[amount]==inf :
            return -1
        else :
            return dp[amount]

进阶,写个二维DP数组的。

python 复制代码
class Solution:
    def coinChange(self, coins: List[int], amount: int) -> int:
        n = len(coins)
        dp = [[inf]*(amount+1) for _ in range(n+1)]
        
        # 二维需要注意的地方真多啊,初始化要考虑的更全面
        dp[0][0] = 0
        for i in range(n+1):
            dp[i][0] = 0
        # 当没有硬币时,想要凑成 amount 是不可能的,所以不是0,而是 inf
        '''
        for i in range(amount+1):
            dp[0][i] = 0
        '''
        for i in range(1,n+1):
            # 注意 j 不是从 coins[i-1] 开始,而是从 1 开始
            for j in range(1,amount+1):   
                # 注意这里必须要有判断
                if j < coins[i-1]:  
                    dp[i][j] = dp[i-1][j]
                else:
                    dp[i][j] = min(dp[i-1][j],dp[i][j-coins[i-1]]+1)

        if dp[-1][-1]==inf :
            return -1
        else :
            return dp[-1][-1]

279 完全平方数

未看解答自己编写的青春版

和上一题差不多,本题考虑到如果物品数组也给 [1,n] 的话,两个循环时间复杂度就是O(n^2)了,那么不如先把可能选择的完全平方数先算出来。

python 复制代码
class Solution:
    def numSquares(self, n: int) -> int:
        items = []
        for i in range(1,int(sqrt(n))+2):
            if i*i <= n :
                items.append(i*i)

        m = len(items)
        dp = [inf]*(n+1)
        dp[0] = 0
        
        for i in range(m):
            for j in range(items[i],n+1):
                dp[j] = min(dp[j],dp[j-items[i]]+1)
        
        return dp[n]

重点

没啥好说的,寻找可选的完全平方数的过程,也可以直接放在双重循环里。

本题依然求得是,最少的个数的数量,所以不管是组合方式还是排列方式,均可以,所以在遍历顺序上不做要求。

代码随想录的代码

先遍历物品, 再遍历背包

python 复制代码
class Solution:
    def numSquares(self, n: int) -> int:
        dp = [float('inf')] * (n + 1)
        dp[0] = 0

        for i in range(1, int(n ** 0.5) + 1):  # 遍历物品
            for j in range(i * i, n + 1):  # 遍历背包
                # 更新凑成数字 j 所需的最少完全平方数数量
                dp[j] = min(dp[j - i * i] + 1, dp[j])

        return dp[n]

我的代码(当天晚上理解后自己编写)

过。

一维DP真简单啊

动规周总结

这周稍微对动态规划有点感觉了。

动规周总结

139 单词拆分

未看解答自己编写的青春版

没思路。

重点

这道题,代码随想录写的解答很好,其给出的回溯算法也要进行学习,相当于回忆知识了。

本题的dp数组的构思,递推公式的推导,初始化的选择,遍历顺序的确定,每一步都不容易,每一步都容易错!

单词拆分

代码随想录的代码

回溯

python 复制代码
class Solution:
    def backtracking(self, s: str, wordSet: set[str], startIndex: int) -> bool:
        # 边界情况:已经遍历到字符串末尾,返回True
        if startIndex >= len(s):
            return True

        # 遍历所有可能的拆分位置
        for i in range(startIndex, len(s)):
            word = s[startIndex:i + 1]  # 截取子串
            if word in wordSet and self.backtracking(s, wordSet, i + 1):
                # 如果截取的子串在字典中,并且后续部分也可以被拆分成单词,返回True
                return True

        # 无法进行有效拆分,返回False
        return False

    def wordBreak(self, s: str, wordDict: List[str]) -> bool:
        wordSet = set(wordDict)  # 转换为哈希集合,提高查找效率
        return self.backtracking(s, wordSet, 0)

DP(版本一)

python 复制代码
class Solution:
    def wordBreak(self, s: str, wordDict: List[str]) -> bool:
        wordSet = set(wordDict)
        n = len(s)
        dp = [False] * (n + 1)  # dp[i] 表示字符串的前 i 个字符是否可以被拆分成单词
        dp[0] = True  # 初始状态,空字符串可以被拆分成单词

        for i in range(1, n + 1): # 遍历背包
            for j in range(i): # 遍历单词
                if dp[j] and s[j:i] in wordSet:
                    dp[i] = True  # 如果 s[0:j] 可以被拆分成单词,并且 s[j:i] 在单词集合中存在,则 s[0:i] 可以被拆分成单词
                    break

        return dp[n]

DP(版本二)

python 复制代码
class Solution:
    def wordBreak(self, s: str, wordDict: List[str]) -> bool:
        dp = [False]*(len(s) + 1)
        dp[0] = True
        # 遍历背包
        for j in range(1, len(s) + 1):
            # 遍历单词
            for word in wordDict:
                if j >= len(word):
                    dp[j] = dp[j] or (dp[j - len(word)] and word == s[j - len(word):j])
        return dp[len(s)]

我的代码(当天晚上理解后自己编写)

下面这个代码很好理解

python 复制代码
class Solution:
    def wordBreak(self, s: str, wordDict: List[str]) -> bool:
        dp = [False]*(len(s) + 1)
        dp[0] = True
        # 遍历背包
        for j in range(1, len(s) + 1):
            # 遍历单词
            for word in wordDict:
                if j >= len(word):
                    dp[j] = dp[j] or (dp[j - len(word)] and word == s[j - len(word):j])
        return dp[len(s)]

二维DP,但是运行错误的代码,从DP数组上看,可以看出错误,但是还没想好怎么改

python 复制代码
class Solution:
    def wordBreak(self, s: str, wordDict: List[str]) -> bool:
        n = len(wordDict)
        dp = [[False]*(len(s) + 1) for _ in range(n+1)]

        dp[0][0] = True
        for i in range(n+1):
            dp[i][0] = True
        
       
        for j in range(1, len(s) + 1):
            for i in range(1,n+1) :
                if j < len(wordDict[i-1]):
                    dp[i][j] = dp[i-1][j]
                else :
                    
                    flag = dp[i][j-len(wordDict[i-1])] and wordDict[i-1] == s[j-len(wordDict[i-1]):j]
                    dp[i][j] = dp[i-1][j] or flag

        return dp[-1][-1]

难道,完全背包+排列,问题,只能用一维DP数组去做?不过这种问题的通性就是,不用刻意往完全背包上套,就从动态规划的角度,设立一维数组,也是能理解代码的。

完全背包+排列,问题,传统意义上的二维DP能不能做?这里存一个疑问,后面再想想,其他的完全背包相关问题,二维DP是可以做的,我在前面也写了。

多重背包理论基础

多加一个循环,遍历物品个数即可,我不喜欢,展开成背包问题的写法。同时,此题也告诉了我们:01背包中的物品,可以是相同的。

多重背包理论基础

代码随想录的代码

python 复制代码
def test_multi_pack():
    weight = [1, 3, 4]
    value = [15, 20, 30]
    nums = [2, 3, 2]
    bagWeight = 10
    dp = [0] * (bagWeight + 1)

    for i in range(len(weight)):  # 遍历物品
        for j in range(bagWeight, weight[i] - 1, -1):  # 遍历背包容量
            # 以上为01背包,然后加一个遍历个数
            for k in range(1, nums[i] + 1):  # 遍历个数
                if j - k * weight[i] >= 0:
                    dp[j] = max(dp[j], dp[j - k * weight[i]] + k * value[i])

        # 打印一下dp数组
        for j in range(bagWeight + 1):
            print(dp[j], end=" ")
        print()

    print(dp[bagWeight])


test_multi_pack()
python 复制代码
def test_multi_pack(weight, value, nums, bagWeight):
    dp = [0] * (bagWeight + 1)

    for i in range(len(weight)):  # 遍历物品
        for j in range(bagWeight, weight[i] - 1, -1):  # 遍历背包容量
            # 以上为01背包,然后加一个遍历个数
            for k in range(1, nums[i] + 1):  # 遍历个数
                if j - k * weight[i] >= 0:
                    dp[j] = max(dp[j], dp[j - k * weight[i]] + k * value[i])

        # 使用 join 函数打印 dp 数组
        print(' '.join(str(dp[j]) for j in range(bagWeight + 1)))

    print(dp[bagWeight])





if __name__ == "__main__":
    weight = [1, 3, 4]
    value = [15, 20, 30]
    nums = [2, 3, 2]
    bagWeight = 10
    test_multi_pack(weight, value, nums, bagWeight)

背包问题总结篇

背包问题结束啦,代码随想录对不同问题进行了不同角度的分类,用来复习真的很棒。
背包问题总结篇

198 打家劫舍

未看解答自己编写的青春版

依旧是按照五部曲来分析的。

dp数组的含义 : 考虑到第 i 个房屋,包含第 i 个房屋,所能偷的最大金额。

python 复制代码
class Solution:
    def rob(self, nums: List[int]) -> int:
        n = len(nums)
        dp = [0]*(n+1)
        dp[1] = nums[0]

        for i in range(2,n+1):
            dp[i] = max(dp[i-1],dp[i-2]+nums[i-1])

        return dp[n]

一点感悟

目前感觉自己做动态规划类的题目,有一个问题就是,对dp数组的含义有非常大的执著,总觉得要将其定义为一个看上去非常合理的量,认为这是求解问题的基础。

但是其实根本不必!就是,题目问什么,就定义什么就好,有的问题再适当做一点改变和限制就好。

dp数组的定义不用太纠结,dp数组定义的很抽象,靠递推关系也可以使状态转移进行的很明朗。

重点

本题其实我多考虑了一个无用状态 dp[0] ,因为在我的定义中,i 是代表第 i 个房屋,而房屋下标的开始就是1. 在代码随想录的讲解中,i 是代表下标为 i 为房屋,这样就可以从0开始了,不过我这样初始化好像更简单一些。

dp[i] 仅仅是考虑到第 i 个房屋,不考虑偷与不偷,偷与不偷,通过递推公式去考虑,而不是代表偷第 i 个房屋。在dp定义中,显示地考虑每个房间偷与不偷状态的数组,为二维dp数组。

值得注意的是

像我对本题dp数组的定义方式,还是不如代码随想录中给出的,这一点在下一题就有所体现,如果所有房屋组成了一个环,那么我就不能在前面硬生生加入一个dp[0]了,还是要按照房屋下标去考虑。

代码随想录的代码

1维DP

python 复制代码
class Solution:
    def rob(self, nums: List[int]) -> int:
        if len(nums) == 0:  # 如果没有房屋,返回0
            return 0
        if len(nums) == 1:  # 如果只有一个房屋,返回其金额
            return nums[0]

        # 创建一个动态规划数组,用于存储最大金额
        dp = [0] * len(nums)
        dp[0] = nums[0]  # 将dp的第一个元素设置为第一个房屋的金额
        dp[1] = max(nums[0], nums[1])  # 将dp的第二个元素设置为第一二个房屋中的金额较大者

        # 遍历剩余的房屋
        for i in range(2, len(nums)):
            # 对于每个房屋,选择抢劫当前房屋和抢劫前一个房屋的最大金额
            dp[i] = max(dp[i - 2] + nums[i], dp[i - 1])

        return dp[-1]  # 返回最后一个房屋中可抢劫的最大金额

2维DP

python 复制代码
class Solution:
    def rob(self, nums: List[int]) -> int:
        if not nums:  # 如果没有房屋,返回0
            return 0

        n = len(nums)
        dp = [[0, 0] for _ in range(n)]  # 创建二维动态规划数组,dp[i][0]表示不抢劫第i个房屋的最大金额,dp[i][1]表示抢劫第i个房屋的最大金额

        dp[0][1] = nums[0]  # 抢劫第一个房屋的最大金额为第一个房屋的金额

        for i in range(1, n):
            dp[i][0] = max(dp[i-1][0], dp[i-1][1])  # 不抢劫第i个房屋,最大金额为前一个房屋抢劫和不抢劫的最大值
            dp[i][1] = dp[i-1][0] + nums[i]  # 抢劫第i个房屋,最大金额为前一个房屋不抢劫的最大金额加上当前房屋的金额

        return max(dp[n-1][0], dp[n-1][1])  # 返回最后一个房屋中可抢劫的最大金额

优化版

python 复制代码
class Solution:
    def rob(self, nums: List[int]) -> int:
        if not nums:  # 如果没有房屋,返回0
            return 0

        prev_max = 0  # 上一个房屋的最大金额
        curr_max = 0  # 当前房屋的最大金额

        for num in nums:
            temp = curr_max  # 临时变量保存当前房屋的最大金额
            curr_max = max(prev_max + num, curr_max)  # 更新当前房屋的最大金额
            prev_max = temp  # 更新上一个房屋的最大金额

        return curr_max  # 返回最后一个房屋中可抢劫的最大金额

我的代码(当天晚上理解后自己编写)

过。

213 打家劫舍II

未看解答自己编写的青春版

没写出来,被环形房屋卡住了,纠结在于怎样处理第0个和第n个之间的关系。

错误代码中的定义:dp[i][0] : 没偷第0个,包括第 i 个房屋,所能获得的最大金币。dp[i][1] : 偷第0个,包括第 i 个房屋,所能获得的最大金币。

python 复制代码
class Solution:
    def rob(self, nums: List[int]) -> int:
        n = len(nums)

        dp = [[0]*2 for _ in range(n)]
        dp[0][0] = 0
        dp[0][1] = nums[0]
        dp[1][0] = nums[1]
        dp[1][1] = nums[0]

        for i in range(2,n):
            dp[i][0] = max(dp[i-1][0],dp[i-2][0]+nums[i])
            dp[i][1] = dp[i-1][1]

        return max(dp[n-1])

我这个错误的代码,打印一下dp数组,就可以明显看出错误。思考了一下应该没得改,这种dp数组的定义方式就是错误的,带有全局意义的变量,不能将某一个元素的特性赋给它。

重点

还是思考方向错了,直接暴力区分,考虑首元素不考虑尾元素,考虑尾元素不考虑首元素,两种逻辑即可。说白了就是上一题的逻辑跑两遍,每次传入的房间数组,分别是去除尾元素和去除首元素的,然后取最大即可。

解决环的问题,分情况讨论。

环形问题,使我们找不到起始位置,要考虑将环形问题展开,展开为线性问题。

代码随想录的代码

python 复制代码
class Solution:
    def rob(self, nums: List[int]) -> int:
        if len(nums) == 0:
            return 0
        if len(nums) == 1:
            return nums[0]
        
        result1 = self.robRange(nums, 0, len(nums) - 2)  # 情况二
        result2 = self.robRange(nums, 1, len(nums) - 1)  # 情况三
        return max(result1, result2)
    # 198.打家劫舍的逻辑
    def robRange(self, nums: List[int], start: int, end: int) -> int:
        if end == start:
            return nums[start]
        
        prev_max = nums[start]
        curr_max = max(nums[start], nums[start + 1])
        
        for i in range(start + 2, end + 1):
            temp = curr_max
            curr_max = max(prev_max + nums[i], curr_max)
            prev_max = temp
        
        return curr_max

2维DP

python 复制代码
class Solution:
    def rob(self, nums: List[int]) -> int:
        if len(nums) < 3:
            return max(nums)

        # 情况二:不抢劫第一个房屋
        result1 = self.robRange(nums[:-1])

        # 情况三:不抢劫最后一个房屋
        result2 = self.robRange(nums[1:])

        return max(result1, result2)

    def robRange(self, nums):
        dp = [[0, 0] for _ in range(len(nums))]
        dp[0][1] = nums[0]

        for i in range(1, len(nums)):
            dp[i][0] = max(dp[i - 1])
            dp[i][1] = dp[i - 1][0] + nums[i]

        return max(dp[-1])

优化版

python 复制代码
class Solution:
    def rob(self, nums: List[int]) -> int:
        if not nums:  # 如果没有房屋,返回0
            return 0

        if len(nums) == 1:  # 如果只有一个房屋,返回该房屋的金额
            return nums[0]

        # 情况二:不抢劫第一个房屋
        prev_max = 0  # 上一个房屋的最大金额
        curr_max = 0  # 当前房屋的最大金额
        for num in nums[1:]:
            temp = curr_max  # 临时变量保存当前房屋的最大金额
            curr_max = max(prev_max + num, curr_max)  # 更新当前房屋的最大金额
            prev_max = temp  # 更新上一个房屋的最大金额
        result1 = curr_max

        # 情况三:不抢劫最后一个房屋
        prev_max = 0  # 上一个房屋的最大金额
        curr_max = 0  # 当前房屋的最大金额
        for num in nums[:-1]:
            temp = curr_max  # 临时变量保存当前房屋的最大金额
            curr_max = max(prev_max + num, curr_max)  # 更新当前房屋的最大金额
            prev_max = temp  # 更新上一个房屋的最大金额
        result2 = curr_max

        return max(result1, result2)

我的代码(当天晚上理解后自己编写)

过。

337 打家劫舍 III

这道题一定要多练习

未看解答自己编写的青春版

超出时间限制,代码思路是代码随想录中的暴力递归版。

python 复制代码
class Solution:
    def rob(self, root: Optional[TreeNode]) -> int:
        if root == None :
            return 0
        if root.left == None and root.right == 0:
            return root.val
       
        left = self.rob(root.left)
        
        right = self.rob(root.right)

        if left == 0 and right == 0 :
            return root.val
        elif left == 0 and right != 0 :
            return max(root.val+self.rob(root.right.left)+self.rob(root.right.right),right)
        elif left != 0 and right == 0 :
            return max(root.val+self.rob(root.left.left)+self.rob(root.left.right),left)
        else :
            return max(root.val+self.rob(root.left.left)+self.rob(root.left.right)+self.rob(root.right.left)+self.rob(root.right.right),left+right)

重点

代码随想录给出的博文中,分析了暴力递归超时的原因,许多节点的状态被重复计算了,进而给出解决方案,用两个维度去记录当前节点的状态!

这是我没想到的。

本题的解答博文中,还涉及了记忆化递归的过程,这个在之前应该是没有讲过的(虽然卡哥说他讲过了),看了看代码,其实就是维护一个字典,来储存已经算过的子串,代码上也很好理解,在递归函数一开始判断那里,加上 if 存在 就 return . 递归结束有了值,就加入字典。

本题是一个树形DP的入门题,要学习!放上代码随想录的解答博文链接。

打家劫舍 III--树形DP

注意,本题的dp数组,其实就是一个一维的数组,而且长度为2,就表示当前节点的状态,偷与不偷,不需要去定义包含所有节点的dp数组,因为在递归过程中,每一次递归都会有一个相应的dp数组存在。

代码随想录的代码

暴力递归

python 复制代码
class Solution:
    def rob(self, root: TreeNode) -> int:
        if root is None:
            return 0
        if root.left is None and root.right  is None:
            return root.val
        # 偷父节点
        val1 = root.val
        if root.left:
            val1 += self.rob(root.left.left) + self.rob(root.left.right)
        if root.right:
            val1 += self.rob(root.right.left) + self.rob(root.right.right)
        # 不偷父节点
        val2 = self.rob(root.left) + self.rob(root.right)
        return max(val1, val2)

记忆化递归

python 复制代码
class Solution:
    memory = {}
    def rob(self, root: TreeNode) -> int:
        if root is None:
            return 0
        if root.left is None and root.right  is None:
            return root.val
        if self.memory.get(root) is not None:
            return self.memory[root]
        # 偷父节点
        val1 = root.val
        if root.left:
            val1 += self.rob(root.left.left) + self.rob(root.left.right)
        if root.right:
            val1 += self.rob(root.right.left) + self.rob(root.right.right)
        # 不偷父节点
        val2 = self.rob(root.left) + self.rob(root.right)
        self.memory[root] = max(val1, val2)
        return max(val1, val2)

动态规划

python 复制代码
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def rob(self, root: Optional[TreeNode]) -> int:
        # dp数组(dp table)以及下标的含义:
        # 1. 下标为 0 记录 **不偷该节点** 所得到的的最大金钱
        # 2. 下标为 1 记录 **偷该节点** 所得到的的最大金钱
        dp = self.traversal(root)
        return max(dp)

    # 要用后序遍历, 因为要通过递归函数的返回值来做下一步计算
    def traversal(self, node):
        
        # 递归终止条件,就是遇到了空节点,那肯定是不偷的
        if not node:
            return (0, 0)

        left = self.traversal(node.left)
        right = self.traversal(node.right)

        # 不偷当前节点, 偷子节点
        # 子节点偷不偷,决定于哪个值更大
        val_0 = max(left[0], left[1]) + max(right[0], right[1])

        # 偷当前节点, 不偷子节点
        val_1 = node.val + left[0] + right[0]

        return (val_0, val_1)

我的代码(当天晚上理解后自己编写)

一定要多练习,这道题。

买卖股票系列,最重要的就是学会这种DP数组定义方式,dp[i][0]表示第i天持有股票的最大现金,dp[i][1]表示第i天不持有股票的最大现金

121 买卖股票的最佳时机

未看解答自己编写的青春版

解答错误,错在无法明确股票是在哪一天买入和卖出的,在递推公式上,就会出现错误和混乱。

错误的dp数组含义:在前 i 天买入卖出所得的最大金钱。

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        dp = [0]*len(prices)
        dp[0]=-inf
        dp[1]=prices[1]-prices[0]
        for i in range(2,len(prices)):
            temp = [dp[i-1],dp[i-1]-prices[i-1]+prices[i],prices[i]-prices[i-1]]
            dp[i] = max(temp)
        print(dp)

        return dp[len(prices)-1]

又想了想,想出了一个自己觉得就很垃圾的方案。

dp数组含义:在第 i 天卖出所得的最大金钱。在本题中,初始化同样需要注意,本题的示例中已说明,如果没有交易完成,return 0 。同样的,要首先判断一下 prices 数组是不是只有一个值,一个值也要return 0。在我的定义下,对于dp[0]可以不做初始化,因为这种情况是非法的,而且我在前面已经做了 if 判断。

我这个方法最后还要对dp数组取一个max

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        if len(prices)==1:
            return 0
        dp = [0]*len(prices)
        
        dp[1]=prices[1]-prices[0]
        for i in range(2,len(prices)):
            temp = [dp[i-1]-prices[i-1]+prices[i],prices[i]-prices[i-1]]
            dp[i] = max(temp)
        

        return max(dp)

重点

本题还有一个很朴素的贪心法。因为股票就买卖一次,那么贪心的想法很自然就是取最左最小值,取最右最大值,那么得到的差值就是最大利润。

跟股票买卖有关的问题,dp数组的定义,基本都是从本题定义的基础上演变过去的。还是要学习代码随想录这一套定义dp数组的方法的,我的那种方法可能只能解决这一道题,不具有普适性。

本题的初始化也是需要理解的。

代码随想录的代码

贪心法:

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        low = float("inf")
        result = 0
        for i in range(len(prices)):
            low = min(low, prices[i]) #取最左最小价格
            result = max(result, prices[i] - low) #直接取最大区间利润
        return result

动态规划:版本一

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        length = len(prices)
        if len == 0:
            return 0
        dp = [[0] * 2 for _ in range(length)]
        dp[0][0] = -prices[0]
        dp[0][1] = 0
        for i in range(1, length):
            dp[i][0] = max(dp[i-1][0], -prices[i])
            dp[i][1] = max(dp[i-1][1], prices[i] + dp[i-1][0])
        return dp[-1][1]

动态规划:版本二

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        length = len(prices)
        dp = [[0] * 2 for _ in range(2)] #注意这里只开辟了一个2 * 2大小的二维数组
        dp[0][0] = -prices[0]
        dp[0][1] = 0
        for i in range(1, length):
            dp[i % 2][0] = max(dp[(i-1) % 2][0], -prices[i])
            dp[i % 2][1] = max(dp[(i-1) % 2][1], prices[i] + dp[(i-1) % 2][0])
        return dp[(length-1) % 2][1]

动态规划:版本三

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        length = len(prices)
        dp0, dp1 = -prices[0], 0 #注意这里只维护两个常量,因为dp0的更新不受dp1的影响
        for i in range(1, length):
            dp1 = max(dp1, dp0 + prices[i])
            dp0 = max(dp0, -prices[i])
        return dp1

我的代码(当天晚上理解后自己编写)

过。

动归周总结

代码随想录的博文总结的很好。

动归周总结

122 买卖股票的最佳时机 II

这道题用上一题我的dp数组定义就行不通了。

所以还是学习代码随想录定义的dp数组含义吧。

未看解答自己编写的青春版

借鉴了上一题,代码随想录的dp数组定义,本题只改变了递推逻辑的部分。

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        length = len(prices)
        if len == 0 or len(prices)==1:
            return 0
        dp = [[0] * 2 for _ in range(length)]
        dp[0][0] = -prices[0]
        dp[0][1] = 0
        for i in range(1,length):
            dp[i][0] = max(dp[i-1][1]-prices[i],dp[i-1][0])
            dp[i][1] = max(dp[i-1][0]+prices[i],dp[i-1][1])

        return dp[length-1][1]

重点

本题是股票可以买卖多次,上一题,股票只能买卖一次。注意体会两题的区别。其他,诸如:dp数组的含义,以及dp数组的初始化问题,都是一样的。

代码随想录的代码

版本一:

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        length = len(prices)
        dp = [[0] * 2 for _ in range(length)]
        dp[0][0] = -prices[0]
        dp[0][1] = 0
        for i in range(1, length):
            dp[i][0] = max(dp[i-1][0], dp[i-1][1] - prices[i]) #注意这里是和121. 买卖股票的最佳时机唯一不同的地方
            dp[i][1] = max(dp[i-1][1], dp[i-1][0] + prices[i])
        return dp[-1][1]

版本二:

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        length = len(prices)
        dp = [[0] * 2 for _ in range(2)] #注意这里只开辟了一个2 * 2大小的二维数组
        dp[0][0] = -prices[0]
        dp[0][1] = 0
        for i in range(1, length):
            dp[i % 2][0] = max(dp[(i-1) % 2][0], dp[(i-1) % 2][1] - prices[i])
            dp[i % 2][1] = max(dp[(i-1) % 2][1], dp[(i-1) % 2][0] + prices[i])
        return dp[(length-1) % 2][1]

我的代码(当天晚上理解后自己编写)

过。

123 买卖股票的最佳时机 III

未看解答自己编写的青春版

依旧根据前面两题的逻辑思维。要注意的是,本来我想在二维数组后面再加一维,来表示当前是第几次购买的状态,但是三维数组可能在声明上不太方便了,于是就按照代码随想录中的做法,把第三维压缩进第二维了,这样写法也挺好的,也很清晰明了。

本题的dp数组定义不需要变化,只要明确定义4个状态就好,需要注意的是初始值的状态,这里我就在第一次提交时,出错了。

就是第0天第二次持有的状态下,所拥有的最大金钱数,应该和第0天第一次持有的状态,相同,均为 -prices[0] 。 表示当天买入-卖出-买入。一开始我这里设置为0了,产生错误。

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:

        n = len(prices)
        if n==0 or n==1:
            return 0
        dp = [[0]*4 for _ in range(n)]
        dp[0][0] = -prices[0]
        dp[0][1] = 0
        dp[0][2] = -prices[0]
        dp[0][3] = 0


        for i in range(1,n):
            dp[i][0] = max(dp[i-1][0],-prices[i])
            dp[i][1] = max(dp[i-1][1],dp[i-1][0]+prices[i])
            dp[i][2] = max(dp[i-1][2],dp[i-1][1]-prices[i])
            dp[i][3] = max(dp[i-1][3],dp[i-1][2]+prices[i])

        

        return max(dp[n-1][1],dp[n-1][3])

重点

就是定义四个状态,从来没想过定义一个0状态。

这里不做什么记录了,放上代码随想录的链接。
买卖股票的最佳时机 III

有一点需要学习的地方:

最大的时候一定是卖出的状态,而两次卖出的状态现金最大一定是最后一次卖出。如果想不明白的录友也可以这么理解:如果第一次卖出已经是最大值了,那么我们可以在当天立刻买入再立刻卖出。所以dp[n][4]已经包含了dp[n][2]的情况。也就是说第二次卖出手里所剩的钱一定是最多的。

代码随想录的代码

版本一:

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        if len(prices) == 0:
            return 0
        dp = [[0] * 5 for _ in range(len(prices))]
        dp[0][1] = -prices[0]
        dp[0][3] = -prices[0]
        for i in range(1, len(prices)):
            dp[i][0] = dp[i-1][0]
            dp[i][1] = max(dp[i-1][1], dp[i-1][0] - prices[i])
            dp[i][2] = max(dp[i-1][2], dp[i-1][1] + prices[i])
            dp[i][3] = max(dp[i-1][3], dp[i-1][2] - prices[i])
            dp[i][4] = max(dp[i-1][4], dp[i-1][3] + prices[i])
        return dp[-1][4]

版本二:

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        if len(prices) == 0:
            return 0
        dp = [0] * 5 
        dp[1] = -prices[0]
        dp[3] = -prices[0]
        for i in range(1, len(prices)):
            dp[1] = max(dp[1], dp[0] - prices[i])
            dp[2] = max(dp[2], dp[1] + prices[i])
            dp[3] = max(dp[3], dp[2] - prices[i])
            dp[4] = max(dp[4], dp[3] + prices[i])
        return dp[4]

我的代码(当天晚上理解后自己编写)

过。

188 买卖股票的最佳时机IV

未看解答自己编写的青春版

同样的,就是上一题的简单拓展,但这里,我觉得我自己对在遍历中的第1次买入,也就是index为0的操作,有些冗余,我是用一个 if 条件去单独判断的。不知道代码随想录中怎么写。

python 复制代码
class Solution:
    def maxProfit(self, k: int, prices: List[int]) -> int:
        n = len(prices)
        if n==0 or n==1:
            return 0
        m = int(k*2)
        dp = [[0]*m for _ in range(n)]

        for i in range(m):
            if i %2 == 0:
                dp[0][i] = -prices[0]


        for i in range(1,n):
            for j in range(m):
                if j % 2 == 0:
                    if j == 0 :
                        dp[i][j] = max(dp[i-1][j],-prices[i])
                    else :
                        dp[i][j] = max(dp[i-1][j],dp[i-1][j-1]-prices[i])
                else :
                    dp[i][j] = max(dp[i-1][j],dp[i-1][j-1]+prices[i])
                

        return dp[n-1][m-1]

重点

代码随想录对上述我提到的问题的解决方法是:多定义一个状态0,贯穿始终,始终不变,一直为0 。

代码随想录的代码

版本一

python 复制代码
class Solution:
    def maxProfit(self, k: int, prices: List[int]) -> int:
        if len(prices) == 0:
            return 0
        dp = [[0] * (2*k+1) for _ in range(len(prices))]
        for j in range(1, 2*k, 2):
            dp[0][j] = -prices[0]
        for i in range(1, len(prices)):
            for j in range(0, 2*k-1, 2):
                dp[i][j+1] = max(dp[i-1][j+1], dp[i-1][j] - prices[i])
                dp[i][j+2] = max(dp[i-1][j+2], dp[i-1][j+1] + prices[i])
        return dp[-1][2*k]

版本二

python 复制代码
class Solution:
    def maxProfit(self, k: int, prices: List[int]) -> int:
        if len(prices) == 0: return 0
        dp = [0] * (2*k + 1)
        for i in range(1,2*k,2):
            dp[i] = -prices[0]
        for i in range(1,len(prices)):
            for j in range(1,2*k + 1):
                if j % 2:
                    dp[j] = max(dp[j],dp[j-1]-prices[i])
                else:
                    dp[j] = max(dp[j],dp[j-1]+prices[i])
        return dp[2*k]

我的代码(当天晚上理解后自己编写)

过。

309 最佳买卖股票时机含冷冻期

买卖股票系列最重要的一道题了,学习四种状态的定义,持有股票,不持有股票但已过冷冻期,不持有股票但在冷冻期(换句话说就是前一天卖的),今天卖股票

未看解答自己编写的青春版

我还是对上一题的思路,稍加修改,先算出了最大的K值,但是这样的后果就是,不管是时间还是占用内存上,均只击败了5%的用户。

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        

        n = len(prices)
        if n==0 or n==1:
            return 0
        k = (n//3)+1
        m = int(k*2)
        dp = [[0]*m for _ in range(n)]

        for i in range(m):
            if i %2 == 0:
                dp[0][i] = -prices[0]


        for i in range(1,n):
            for j in range(m):
                if j % 2 == 0:
                    if j == 0 :
                        dp[i][j] = max(dp[i-1][j],-prices[i])
                    else :
                        dp[i][j] = max(dp[i-1][j],dp[i-2][j-1]-prices[i])
                else :
                    dp[i][j] = max(dp[i-1][j],dp[i-1][j-1]+prices[i])
                

        return dp[n-1][m-1]

重点

买卖K次那个题没办法,必须定义那么多状态,但是对于本题,其实只要定义4个状态就可以了,去学习代码随想录的题解。

要灵活啊,谁说买卖多次就必须要有循环了,只有最多买卖K次,需要再套一层循环,前面也做了无限次买卖的题了,不也只有外面一层循环?



代码随想录的代码

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        n = len(prices)
        if n == 0:
            return 0
        dp = [[0] * 4 for _ in range(n)]
        dp[0][0] = -prices[0] #持股票
        for i in range(1, n):
            dp[i][0] = max(dp[i-1][0], max(dp[i-1][3], dp[i-1][1]) - prices[i])
            dp[i][1] = max(dp[i-1][1], dp[i-1][3])
            dp[i][2] = dp[i-1][0] + prices[i]
            dp[i][3] = dp[i-1][2]
        return max(dp[n-1][3], dp[n-1][1], dp[n-1][2])

我的代码(当天晚上理解后自己编写)

过。

动规周总结

四道买卖股票题目总结。

动规周总结

714 买卖股票的最佳时机含手续费

未看解答自己编写的青春版

在卖股票的时候,把手续费减去就好了,注意初始化那里,第0天不持有股票依然返回0,不买不卖,而不是一个负数,因为求的是最大值。

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int], fee: int) -> int:
        length = len(prices)
        if len == 0 or len(prices)==1:
            return 0
        dp = [[0] * 2 for _ in range(length)]
        dp[0][0] = -prices[0]
        dp[0][1] = 0
        for i in range(1,length):
            dp[i][0] = max(dp[i-1][1]-prices[i],dp[i-1][0])
            dp[i][1] = max(dp[i-1][0]+prices[i]-fee,dp[i-1][1])

        return dp[length-1][1]

重点

这道题要掌握的方法,还有一个贪心法,动态规划方法没什么新意,但是不知道为什么,这道题贪心法的解答博客,竟然没有在话题目录里。

贪心法还要理解一下,时间原因没细看。

动态规划思路

贪心思路

代码随想录的代码

动态规划:

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int], fee: int) -> int:
        n = len(prices)
        dp = [[0] * 2 for _ in range(n)]
        dp[0][0] = -prices[0] #持股票
        for i in range(1, n):
            dp[i][0] = max(dp[i-1][0], dp[i-1][1] - prices[i])
            dp[i][1] = max(dp[i-1][1], dp[i-1][0] + prices[i] - fee)
        return max(dp[-1][0], dp[-1][1])

贪心法:

python 复制代码
class Solution: # 贪心思路
    def maxProfit(self, prices: List[int], fee: int) -> int:
        result = 0
        minPrice = prices[0]
        for i in range(1, len(prices)):
            if prices[i] < minPrice: # 此时有更低的价格,可以买入
                minPrice = prices[i]
            elif prices[i] > (minPrice + fee): # 此时有利润,同时假买入高价的股票,看看是否继续盈利
                result += prices[i] - (minPrice + fee)
                minPrice = prices[i] - fee
            else: # minPrice<= prices[i] <= minPrice + fee, 价格处于minPrice和minPrice+fee之间,不做操作
                continue
        return result

我的代码(当天晚上理解后自己编写)

关于股票问题,想想清楚,对于限制,只可买卖一次,和不限制买卖次数的题目来说,dp数组的定义可以是相同的,就定义两个状态,dp[i][0]:持有股票,dp[i][1]:不持有股票,都对第0天做单独初始化,唯一区别在于,在循环中推导 dp[i][0]时,需不需要前一天的状态。

对于限制最多交易次数的题,就按照顺序,去定义每一天的状态。

含冷冻期的题,注意定义的四种状态。

股票问题总结篇

股票问题总结篇

相关推荐
哭泣的眼泪4081 小时前
解析粗糙度仪在工业制造及材料科学和建筑工程领域的重要性
python·算法·django·virtualenv·pygame
Microsoft Word2 小时前
c++基础语法
开发语言·c++·算法
天才在此2 小时前
汽车加油行驶问题-动态规划算法(已在洛谷AC)
算法·动态规划
莫叫石榴姐3 小时前
数据科学与SQL:组距分组分析 | 区间分布问题
大数据·人工智能·sql·深度学习·算法·机器学习·数据挖掘
茶猫_4 小时前
力扣面试题 - 25 二进制数转字符串
c语言·算法·leetcode·职场和发展
肥猪猪爸6 小时前
使用卡尔曼滤波器估计pybullet中的机器人位置
数据结构·人工智能·python·算法·机器人·卡尔曼滤波·pybullet
readmancynn6 小时前
二分基本实现
数据结构·算法
萝卜兽编程6 小时前
优先级队列
c++·算法
盼海7 小时前
排序算法(四)--快速排序
数据结构·算法·排序算法
一直学习永不止步7 小时前
LeetCode题练习与总结:最长回文串--409
java·数据结构·算法·leetcode·字符串·贪心·哈希表