清单
● 123.买卖股票的最佳时机III
● 188.买卖股票的最佳时机IV
LeetCode #123 买卖股票的最佳时机 III
1. 题目
给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成两笔交易。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
2. 思路
- dp数组含义: 最多完成两笔交易,共存在五种状态 1.不交易 2. 第一次买入 3. 第一次卖出 4. 第二次买入 5. 第二次卖出 dp[i][j] 为第i天所能获取的最大利润, i 代表天数 j代表交易状态
- 递推公式:
- 若i天均不操作: dp[i][0] = dp[i-1][0]
- 若i天第一次买入: dp[i][1] = max(dp[i-1][1] #前一天已买入, dp[i-1][0] - prices[i] # i 天买入
- 第i天第一次卖出: dp[i][2] = max(dp[i-1][2] #前一天已卖出, dp[i-1][1] + prices[i] # i 天卖出
- 若i天第二次买入: dp[i][3] = max(dp[i-1][3] #前一天已买入, dp[i-1][2] - prices[i] # i 天买入
- 第i天第二次卖出: dp[i][4] = max(dp[i-1][4] #前一天已卖出, dp[i-1][3] + prices[i] # i 天卖出
- 初始化: dp[0][1] = - prices[i] dp[0][3] = - prices[i]
- 遍历顺序: 正向遍历
3. 代码实现
class Solution:def maxProfit(self, prices: List[int]) -> int: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]
LeetCode #188 买卖股票的最佳时机 IV
1. 题目
给定一个整数数组 prices ,它的第 i 个元素 prices[i] 是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
2. 思路
思路同上一题一样,共存在2k+1种状态 0为不操作,奇数为买入,偶数为卖出
3. 代码实现
class Solution:def maxProfit(self, k: int, prices: List[int]) -> int:dp = [[0] * ( 2 * k + 1) for _ in range(len(prices))]for i in range(1, 2 * k, 2):dp[0][i] = - 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]