本题是123. 买卖股票的最佳时机 III的拓展
首先排除股票天数或k为0的边界条件.
有k次交易, 所以有2*k个持有/空仓状态. 第k个持有状态代表当前持有股票, 且进行了k次买入.
- 每一天的第k个持有状态可以由第k-1个空仓状态, 或前一天的原状态得来.
- 注意, k=0时为边界条件, 可以来自最开始空仓状态的0收益, 或前一天的原状态.
- 每一天的第k个空仓状态可以由第k-1个持有状态, 或前一天的原状态得来.
最后返回第k个空仓状态, 不用返回所有空仓状态的最大值是因为它可以由之前的空仓状态推导而来.
class Solution(object):
def maxProfit(self, k, prices):
"""
:type k: int
:type prices: List[int]
:rtype: int
"""
if len(prices) == 0 or k == 0:
return 0
dp = [-prices[0] if i % 2 == 0 else 0 for i in range(k*2)]
n = len(prices)
for i in range(1, n):
for j in range(k*2):
if j == 0:
dp[j] = max(dp[j], -prices[i])
elif j % 2 == 0:
dp[j] = max(dp[j], dp[j-1] - prices[i])
else:
dp[j] = max(dp[j], dp[j-1] + prices[i])
return dp[k*2-1]