leetcode刷题-动态规划08

代码随想录动态规划part08|121. 买卖股票的最佳时机、122.买卖股票的最佳时机II、123.买卖股票的最佳时机III

121.买卖股票的最佳时机

leetcode题目链接
代码随想录文档讲解

思路

本题这一只股票只买卖一次

暴力解法、动态规划解法

动态规划:

  1. dp数组的含义:
    要用二维dp数组:dp[i][0], dp[i][1]分别表示持有这只股票所得的最大现金和不持有这支股票的最大现金(刚开始手里的钱数为0,那最终的钱数就是利润)。
    最终返回max(dp[len(nums-1)][0], dp[len(nums)-1][1]),其实是返回dp[len(nums)-1][1],因为最后一天不持有股票手里的现金才会多
  2. 递推公式:
    a. dp[i][0](第i天已经持有这只股票,可能是之前就持有,也可能是今天才买入持有)
    dp[i][0] = max(dp[i-1][0], -price[i]) (这里负数的max就是花费最少,当前手里的现金肯定是负数)
    b. dp[i][1](第i天已经不持有这只股票,可能是之前就卖了,也可能是今天才卖掉这只股票)
    dp[i][1] = max(dp[i-1][1], dp[i-1][0]+price[i])
    c. 思路比较不寻常,注意别想复杂了,准
  3. dp数组初始化:
    dp[0][0] = -price[0]
    dp[0][1] = 0
  4. 遍历顺序:从前往后遍历,不涉及背包中复杂的遍历顺序

python代码

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

以前的提交:贪心算法

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        maxprofit = 0
        mincost = 90000
        for i in range(len(prices)):
            if prices[i]<mincost:
                mincost = prices[i]
            maxprofit = max(maxprofit, prices[i]-mincost)
        return maxprofit

122.买卖股票的最佳时机II

leetcode题目链接
代码随想录文档讲解

思路

可用贪心法,本题相比上题改为:可买卖多次

贪心算法:

可买卖多次,第i天买入股票手头上的现金不是0,可能是之前买卖多次获得的利润,所以在递推公式中dp[i][0]需要改变

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])

python代码

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        dp = [[0]*2 for _ in range(len(prices))]
        dp[0][0] = -prices[0]
        for i in range(1, len(prices)):
            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])
        return dp[len(prices)-1][1]

123.买卖股票的最佳时机III -- 困难

leetcode题目链接
代码随想录文档讲解

思路

本题最多可以完成两笔交易,不能同时参与多笔交易

  1. dp数组定义
    dp[i][0] 不操作(没有持有过)
    dp[i][1] 第一次持有(第一次买入或者延续了前些天买入后持有的状态)
    dp[i][2] 第一次不持有 (已经卖出了)
    dp[i][3] 第二次持有
    dp[i][4] 第二次不持有
  2. 递推公式
    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])
  3. 初始化
    dp[0][0] = 0
    dp[0][1] = -prices[0]
    dp[0][2] = 0
    dp[0][3] = -prices[0]
    dp[0][4] = 0
  4. 遍历顺序

python代码

python 复制代码
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        dp = [[0]*5 for _ in range(len(prices))]
        dp[0][1], dp[0][3] = -prices[0], -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[len(prices)-1][4] 

本题可以不用维护一个dp数组,只用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]
相关推荐
水水沝淼㵘2 分钟前
嵌入式开发学习日志(数据结构--单链表)Day20
c语言·开发语言·数据结构·学习·算法
算法给的安全感4 分钟前
bfs-最小步数问题
java·算法·宽度优先
灏瀚星空18 分钟前
地磁-惯性-视觉融合制导系统设计:现代空战导航的抗干扰解决方案
图像处理·人工智能·python·深度学习·算法·机器学习·信息与通信
田梓燊36 分钟前
专业课复习笔记 7
笔记·算法
健康胡1 小时前
仿射变换 与 透视变换
图像处理·人工智能·深度学习·opencv·算法·机器学习·计算机视觉
L_cl1 小时前
【Python 算法零基础 2.模拟 ④ 基于矩阵】
python·算法·矩阵
2301_807611491 小时前
310. 最小高度树
c++·算法·leetcode·深度优先·回溯
@ chen1 小时前
常见排序算法及其java实现
java·算法·排序算法
学习中的码虫2 小时前
数据结构中的高级排序算法
数据结构·算法·排序算法
山北雨夜漫步2 小时前
机器学习 Day17 朴素贝叶斯算法-----概率论知识
人工智能·算法·机器学习