LeetCode 309. Best Time to Buy and Sell Stock with Cooldown

LeetCode 309

考虑三种状态的转换，买入，卖出，cooldown，买入之后可以保持不动继续保持买入的状态，卖出之后进入cooldown，在cooldown也可以保持不变。cooldown之后可以买入。根据这个状态转换我们可以有这样的公式
buy【i】 = min（之前买的 buy【i-1】，cooldown之后买的 price【i】 - cooldown【i-1】）
sell【i】= price【i】-buy【i-1】（卖出股票）
cooldown【i】 = max（sell【i-1】卖出之后进入cooldown，cooldown【i-1】保持原来状态）

转为代码：

    def maxProfit(self, prices: List[int]) -> int:        
        l = len(prices)        
        if l == 0: return 0

        buy = [0]*l
        sell = [0] * l
        cooldown = [0]* l
        buy[0] = prices[0]
        for i in range(1, l):
            buy[i] = min(buy[i-1], prices[i] - cooldown[i-1])
            sell[i] = - buy[i-1] + prices[i]
            cooldown[i] = max(sell[i-1], cooldown[i-1])

        return max(sell[l-1], cooldown[l-1])