123.买卖股票的最佳时机III
题目链接:https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-iii/
因为题目定义最多交易为2次 那么二维数组的维度可以定义为0-4,即分为没有操作,第一次买入,第一次卖出,第二次买入,第二次卖出五种状态
但很容易晕的一点是在做公式推导时 买入并不一定代表买入 只是推导此状态,没有想明白这点就很容易晕眩
class Solution:
def maxProfit(self, prices: List[int]) -> int:
#dp[i][j]表示第i天状态j所剩最大金额
#j: 0-4,0为没有买入,1为第一次买入,2为第一次卖出,3为第二次买入,4为第二次卖出
#d[i][1] = max(dp[i-1][0] - prices[i], dp[i-1][1])
#d[i][2] = max(dp[i-1][1] + prices[i], dp[i-1][2])
#d[i][3] = max(dp[i-1][2] - prices[i], dp[i-1][3])
#d[i][4] = max(dp[i-1][3] + prices[i], dp[i-1][4])
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][0] - prices[i], dp[i-1][1])
dp[i][2] = max(dp[i-1][1] + prices[i], dp[i-1][2])
dp[i][3] = max(dp[i-1][2] - prices[i], dp[i-1][3])
dp[i][4] = max(dp[i-1][3] + prices[i], dp[i-1][4])
return dp[-1][4]
188.买卖股票的最佳时机IV
题目链接:https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-iv/
没吃透没吃透没吃透!重要的事情说三遍