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

题目

给定一个整数数组prices，其中第 prices[i] 表示第 i 天的股票价格 。​

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

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

示例

输入: prices = [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]

解析

上一道题还是hard呢，这一道就成了middle了。。。
本题主要是增加了冷冻期的这个概念，从而导致“状态”这个很难想明白，动规五部曲走一下：
1.确定dp数组极其含义
首先还是需要定义一个二维数组dp[i][]j，然后有如下的状态（状态0123）
2.确定递推公式
状态0：处于买入股票状态dp[i][0]；
那么要达到这个状态，需要从前一个状态来进行递推，有如下两种操作：
操作一：前一天就已经买入了，今天继续保持，则dp[i][0] = dp[i-1][0]
操作二：首先是要今天买入（-prices[i]），但前一天的状态可能有两种：
1.前一天是保持卖出状态（状态1），即大前天卖出，前天是冷冻期，昨天属于保持卖出状态
2.前一天是冷冻期状态（状态3）
所以这种情况的递推公式为：dp[i][0] = max(dp[i - 1][0], dp[i - 1][3] - prices[i], dp[i - 1][1] - prices[i]);
状态1：处于保持卖出状态：
有两种可能：
1.前一天就是保持卖出状态
2.前一天是冷冻期
所以此时递推公式为：dp[i][1] = max(dp[i - 1][1], dp[i - 1][3]);
状态2：处于今天卖出状态：
只有一种可能，昨天持有，今天卖出：dp[i][2] = dp[i - 1][0] + prices[i];
状态3：处于冷冻期：
也只有一种可能，昨天卖出，今天冷冻：dp[i][3] = dp[i - 1][2];
3.初始化dp数组
只有买入的那种状态需要初始化为dp[0][0] = -prices[0],其余的默认初始化为0

func maxProfit(prices []int) int {
    n := len(prices)
    dp := make([][]int, n)
    for i := 0; i < n; i++ {
        dp[i] = make([]int, 4) // 0123四种状态
    }
    dp[0][0] = -prices[0] // 只有第1天的买入情况需要初始化，其余的默认为0
    for i := 1; i < n; i++ { // 从第二天开始遍历
        dp[i][0] = max(dp[i-1][0], max(dp[i-1][1] - prices[i], dp[i-1][3] - prices[i]))
        dp[i][1] = max(dp[i-1][1], dp[i-1][3])
        dp[i][2] = dp[i-1][0] + prices[i]
        dp[i][3] = dp[i-1][2]
    }
    return max(dp[n-1][1], max(dp[n-1][2], dp[n-1][3]))
}
func max(a, b int) int {
    if a > b {
        return a
    }
    return b
}