题目链接: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
class Solution:
def maxProfit(self, k: int, prices: List[int]) -> int:
if k == 0 or prices == []:
return 0
n = len(prices)
r = [[0] * (2*k+1) for _ in range(n)]
for i in range(k):
r[0][2*i+1] = -prices[0]
for i in range(1, n):
for j in range(k):
# 第j次买入
r[i][2*j+1] = max(r[i-1][2*j+1], r[i-1][2*j] - prices[i])
# 第j次卖出
r[i][2*j+2] = max(r[i-1][2*j+2], r[i-1][2*j+1] + prices[i])
return r[n-1][2*k]