309.最佳买卖股票时机含冷冻期
含有冷冻期之后,需要dp数组再多加一个状态即冷冻期,如果这一天是冷冻期的最大收益是多少。
class Solution:
def maxProfit(self, prices: List[int]) -> int:
# 不处理,持有,不持有,冷冻期(就是不处理)
N = len(prices)
dp = [[0]*4 for i in range(N)]
dp[0][0] = 0
dp[0][1] = -prices[0]
dp[0][2] = 0
dp[0][3] = 0
for i in range(1,N):
# 持有股票
# 如果此前就持有该股票:dp[i-1][1]
# 如果此前未持有该股票新购入,需要满足前一天冷冻期,今天买入;继续保持前一天持有状态
dp[i][1] = max(dp[i-1][3] - prices[i] ,dp[i-1][1])
dp[i][2] = max(dp[i-1][1] + prices[i] ,dp[i-1][2])
dp[i][3] = dp[i-1][2]
return max(dp[N-1])
714.买卖股票的最佳时机含手续费
类似于买卖股票II,在卖出股票的时候将手续费减去即可。
class Solution:
def maxProfit(self, prices: List[int], fee: int) -> int:
N = len(prices)
dp = [[0,0] for _ in range(N)]
dp[0][0] = -prices[0]
dp[0][1] = 0
for i in range(1,N):
# 0表示不持有该股票;1表示持有该股票
# 如果不持有该股票,一种情况是买入,另一种情况是继续保持不持有的状态
# 如果当前是持有该股票的状态,一种情况是卖出,另一种情况是继续保持之前的持有状态,不卖出
dp[i][0] = max(dp[i-1][1]-prices[i],dp[i-1][0])
dp[i][1] = max(dp[i-1][1],dp[i-1][0] + prices[i] - fee)
return max(dp[N-1][0],dp[N-1][1])