[Swift]LeetCode309. 最佳买卖股票时机含冷冻期 | Best Time to Buy and Sell Stock with Cooldown

★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★
➤微信公众号：山青咏芝（shanqingyongzhi）
➤博客园地址：山青咏芝（https://www.cnblogs.com/strengthen/）
➤GitHub地址：https://github.com/strengthen/LeetCode
➤原文地址：https://www.cnblogs.com/strengthen/p/10260419.html 
➤如果链接不是山青咏芝的博客园地址，则可能是爬取作者的文章。
➤原文已修改更新！强烈建议点击原文地址阅读！支持作者！支持原创！
★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★

Say you have an array for which the ith element is the price of a given stock on day i.

Design an algorithm to find the maximum profit. You may complete as many transactions as you like (ie, buy one and sell one share of the stock multiple times) with the following restrictions:

• You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).
• After you sell your stock, you cannot buy stock on next day. (ie, cooldown 1 day)

Example:

Input: [1,2,3,0,2]
Output: 3 
Explanation: transactions = [buy, sell, cooldown, buy, sell]

---

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

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

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

示例:

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

---

16ms

class Solution {
    func maxProfit(_ prices: [Int]) -> Int {
    if prices.count <= 1 {
         return 0 
    }
    var s0: Int = 0
    var s1: Int = -prices[0]
    var s2: Int = Int.min
    for i in 1 ..< prices.count {
        let pre0 = s0
        let pre1 = s1
        let pre2 = s2
        s0 = max(pre0, pre2)
        s1 = max(pre0 - prices[i], pre1)
        s2 = pre1 + prices[i]
    }
    return max(s0, s2)
}
}

---

28ms

class Solution {
    func maxProfit(_ prices: [Int]) -> Int {
        var buy = Int.min, noOp = Int.min
        var coolDown = 0, sell = 0
        for  p in prices {
            noOp = max(noOp, buy)
            buy = coolDown - p
            coolDown = max(coolDown, sell)
            sell = noOp + p;
        }
        return max(coolDown, sell)
    }
}