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

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

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

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

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

考察重点：此类问题应首先想到动态规划。本题每一天涉及三个状态——买入、冷冻、卖出，所以需要三个数组分别对应三个状态转移方程。其中，buy、cold两数组考虑购入后和卖出后的持续持有或持续观望，因此写出转移方程，max(cold[i-1] - prices[i], buy[i-1]). max(cold[i-1], sell[i-1])；sell数组直接写为buy[i-1] + prices[i]，即卖出操作只存在于买入操作之后。

func maxProfit(prices []int) int {
    pLen := len(prices)
    buy, cold, sell := make([]int, pLen), make([]int, pLen), make([]int, pLen)
    buy[0] = -prices[0]
    cold[0] = 0
    sell[0] = 0
    for i := 1;i < pLen;i ++{
        buy[i] = max(cold[i-1] - prices[i], buy[i-1])
        cold[i] = max(cold[i-1], sell[i-1])
        sell[i] = buy[i-1] + prices[i]
    }
    return max(max(buy[pLen-1], cold[pLen-1]), sell[pLen-1])
}

func max(a, b int)int{
    if a > b{
        return a
    }
    return b
}