题目链接:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-iv/
难度:困难

描述:
给定一个整数数组 prices ,它的第 i 个元素 prices[i] 是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。

提示:
k: [0, 100]
prices.length: [0, 1000]

题解

思路:
与123. 买卖股票的最佳时机 III类似,不过是把2扩展到k

  1. class Solution:
  2. def maxProfit(self, k: int, prices: List[int]) -> int:
  3. if k == 0 or prices == []:
  4. return 0
  5. n = len(prices)
  6. r = [[0] * (2*k+1) for _ in range(n)]
  7. for i in range(k):
  8. r[0][2*i+1] = -prices[0]
  9. for i in range(1, n):
  10. for j in range(k):
  11. # 第j次买入
  12. r[i][2*j+1] = max(r[i-1][2*j+1], r[i-1][2*j] - prices[i])
  13. # 第j次卖出
  14. r[i][2*j+2] = max(r[i-1][2*j+2], r[i-1][2*j+1] + prices[i])
  15. return r[n-1][2*k]