leetcode188买卖股票的时机Ⅳ

难度：困难

题目描述：力扣

思路：动态规划

1）定义状态

buy[i][j]:到第i天，进行j笔交易，并且持有一支股票的最大利润；

sell[i][j]:到第i天，进行j笔交易，并且手上不持有股票的最大利润。

2）状态转移方程

第i-1天持有股票，或者第i-1天不持有股票，第i天买入。

buy[i][j]=max{buy[i−1][j],sell[i−1][j]−price[i]}

第i-1天不持有股票，或者第i-1天持有股票，第i天卖出。

sell[i][j]=max{sell[i−1][j],buy[i−1][j−1]+price[i]}

最大利润为sell[n−1][0..k] 中的最大值。

3）初始状态/边界条件

buy[0][0]:-prices[0];buy[0][1...k]:不合法，设置为float("-inf")

sell[0][0]：0；sell[0][1...k]:不合法，设置为float("-inf")

代码

class Solution:
    def maxProfit(self, k: int, prices: List[int]) -> int:
        if not prices:
            return 0

        n = len(prices)
        k = min(k, n // 2)
        buy = [[0] * (k + 1) for _ in range(n)]
        sell = [[0] * (k + 1) for _ in range(n)]

        buy[0][0], sell[0][0] = -prices[0], 0
        for i in range(1, k + 1):
            buy[0][i] = sell[0][i] = float("-inf")

        for i in range(1, n):
            buy[i][0] = max(buy[i - 1][0], sell[i - 1][0] - prices[i])
            for j in range(1, k + 1):
                buy[i][j] = max(buy[i - 1][j], sell[i - 1][j] - prices[i])
                sell[i][j] = max(sell[i - 1][j], buy[i - 1][j - 1] + prices[i]);  

        return max(sell[n - 1])

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-iv/solution/mai-mai-gu-piao-de-zui-jia-shi-ji-iv-by-8xtkp/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

股票问题系列通解：力扣