[leetcode] 309. Best Time to Buy and Sell Stock with Cooldown

https://leetcode.com/problems/best-time-to-buy-and-sell-stock-with-cooldown/description/

Best Time to Buy and Sell Stock with Cooldown - LeetCode

https://www.youtube.com/watch?v=IGIe46xw3YY 

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        def f(i, buy, prices, size, dp):
            if i >= size:
                return 0
            if dp[i][buy] != -1:
                return dp[i][buy]
            
            if buy == True:
                dp[i][buy] = max(
                        -prices[i] + f(i+1, False, prices, size, dp),
                        0 + f(i+1, True, prices, size, dp)
                    )
                return dp[i][buy]

            dp[i][buy] = max(
                        prices[i] + f(i+2, True, prices, size, dp),
                        0 + f(i+1, False, prices, size, dp)
                    )
            return dp[i][buy]
        
        size = len(prices)
        dp = []
        for i in range(size):
            dp.append({
                True: -1,
                False: -1
            })

        return f(0, True, prices, size, dp)