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]
相关推荐
int型码农4 小时前
数据结构第八章(一) 插入排序
c语言·数据结构·算法·排序算法·希尔排序
UFIT5 小时前
NoSQL之redis哨兵
java·前端·算法
喜欢吃燃面5 小时前
C++刷题:日期模拟(1)
c++·学习·算法
SHERlocked935 小时前
CPP 从 0 到 1 完成一个支持 future/promise 的 Windows 异步串口通信库
c++·算法·promise
怀旧,5 小时前
【数据结构】6. 时间与空间复杂度
java·数据结构·算法
积极向上的向日葵5 小时前
有效的括号题解
数据结构·算法·
GIS小天5 小时前
AI+预测3D新模型百十个定位预测+胆码预测+去和尾2025年6月7日第101弹
人工智能·算法·机器学习·彩票
_Itachi__6 小时前
LeetCode 热题 100 74. 搜索二维矩阵
算法·leetcode·矩阵
不忘不弃6 小时前
计算矩阵A和B的乘积
线性代数·算法·矩阵
不爱写代码的玉子6 小时前
HALCON透视矩阵
人工智能·深度学习·线性代数·算法·计算机视觉·矩阵·c#