代码随想录算法训练营第五十一天

309. 最佳买卖股票时机含冷冻期

题目链接：最佳买卖股票时机含冷冻期
得理清楚冷冻期出现的位置，我一开始是想让dp[i][3]当做查询是否冷冻期的状态的，但是这样太贪心了，没有办法得到最大利润，所以最后还是决定吧不持有不可买和不持有可买分开。

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        # dp[i][0] = 第i天没操作
        # dp[i][1] = 第i天持有
        # dp[i][2] = 第i天不持有但可买
        # dp[i][3] = 第i天不持有但不可买
        dp = [[0]*4 for _ in range(len(prices))]
        dp[0][1] = -prices[0]
        for i in range(1,len(prices)):
            # 第i-1天已经持有
            # 第i-1天卖了冷冻期了
            # 第i-1天没有持有也不是冷冻期，第i天买了
            dp[i][1] = max(dp[i-1][1], dp[i-1][2]-prices[i])
            # 第i-1天已经不持有可买
            # 第i-1天已经不持有但不可买
            dp[i][2] = max(dp[i-1][2],dp[i-1][3])
            # 第i-1天持有卖了,所以第i天不持有也不能买
            dp[i][3] = dp[i-1][1]+prices[i]
        
        return max(dp[-1][2],dp[-1][3])      

714. 买卖股票的最佳时机含手续费

题目链接：买卖股票的最佳时机含手续费
和II是一样的。

class Solution:
    def maxProfit(self, prices: List[int], fee: 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]-fee)
        return dp[-1][-1]