股票买卖问题-含冷冻期

问题

给定一个整数数组，其中第 i 个元素代表了第 i 天的股票价格 。

设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:

你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。

---

问题分析
无限次交易，卖出股票后，第二天无法继续交易。
状态：持有股票；不持有股票，且处于冷冻期；不持有股票，且不在冷冻期
dp[days][3]
dp[i][0] 表示第i天持有股票
dp[i][1]表示第i天不持有股票，冷冻期
dp[i][2]表示第i天不持有股票，不在冷冻期

转化关系：
不持有股票，且不在冷冻期—>持有股票
持有股票—>不持有股票，冷冻期
不持有股票，冷冻期—>不持有股票，不在冷冻期

---

代码

def stock5(prices):
    
    if len(prices)<2:
        return 0
    
    dp = [[0]*3 for _ in range(len(prices))]
    #dp[i][0] 持有股票
    #dp[i][1] 不持有股票，冷冻期
    #dp[i][2] 不持有股票，非冷冻期
    
    dp[0][0] = -prices[0]
    
    for i in range(1,len(prices)):
        
        dp[i][0] = max(dp[i-1][0],dp[i-1][2]-prices[i])
        dp[i][1] = dp[i-1][0] + prices[i]
        dp[i][2] = max(dp[i-1][2],dp[i-1][1])
    
    return max(dp[len(prices)-1][1],dp[len(prices)-1][2])