思路

定义状态 状态转移 初始情况

实现

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        # 先想想状态的转移 买入->卖出 卖出->冷冻->买入 买入->持有->卖出
        # 定义状态 dp[i][k] = p 第i天的状态为k时最大的利润p
        # k=0 为买入且持有 k=1为卖出 k=2为没有股票 k=3为冷冻
        # 转移 dp[i][0] = max(dp[i-2][1],dp[i-1][3],dp[i-1][2])  dp[i][1]=max(dp[i-1][0],dp[i-1][2])
        # dp[i][2] = max(dp[i-1][1],dp[i-1][2]) dp[i][3] = dp[i-1][1]
        p_len =len(prices)
        if not p_len: # 没有天
            return 0
        if p_len==1: # 只有一天
            return 0
        # 大于等于俩天
        dp = [[0]*4 for _ in range(p_len)]
        # 初始态 第一天肯定除了买入都是0 第二天 如果可以把第一天的卖出 有赚钱就更新一下状态1的值
        dp[0][0] = -prices[0]
        for i in range(1,p_len):
            dp[i][0] = max(dp[i-1][3]-prices[i],dp[i-1][2]-prices[i],dp[i-1][0])
            dp[i][1] = dp[i-1][0]+prices[i]
            dp[i][2] = max(dp[i-1][1],dp[i-1][2])
            dp[i][3] = dp[i-1][1]
        print(dp)
        return max(dp[p_len-1][0],dp[p_len-1][1],dp[p_len-1][2],dp[p_len-1][3])

优化