今日任务
目录
123.买卖股票的最佳时机III - Hard
给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
提示:至多买卖两次,意味着可以买卖一次,可以买卖两次,也可以不买卖;那么一天一共会有五个状态:
- 没有操作 (其实我们也可以不设置这个状态)
- 第一次持有股票
- 第一次不持有股票
- 第二次持有股票
- 第二次不持有股票
dp[i][j]中 i表示第i天,j为 [0 - 4] 五个状态,dp[i][j]表示第i天状态j所剩最大现金
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][0] = 0
dp[0][1] = -prices[0]
dp[0][2] = 0
dp[0][3] = -prices[0]
dp[0][4] = 0
for i in range(1, len(prices)):
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 max(dp[-1][2], dp[-1][4])
188.买卖股票的最佳时机IV - Hard
题目链接:力扣-188. 买卖股票的最佳时机 IV
给定一个整数数组 prices ,它的第 i 个元素 prices[i] 是一支给定的股票在第 i 天的价格,和一个整型 k 。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。也就是说,你最多可以买 k 次,卖 k 次。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
提示:将上题的代码泛化
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 i in range(2*k + 1):
if i % 2 == 0:
dp[0][i] = 0
else:
dp[0][i] = -prices[0]
for i in range(1, len(prices)):
for j in range(1, 2*k + 1):
if j % 2 == 1:
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 max(dp[-1])