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] = 0sell[0] = 0for 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
}