问题
给定一个整数数组 prices ,它的第 i 个元素 prices[i] 是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
问题分析
类似于问题:股票买卖问题-最多2次交易
状态变化有:
持有股票:0次持有,第1次持有,第2次持有,…,第k次持有
不持有股票:0次不持有,第1次不持有,第2次不持有,…,第k次不持有
dp[days][2][k+1]
dp[i][0][j]表示第i天,第j次持有股票时的最大利润
dp[i][1][j]表示第i天,第j次不持有股票时的最大利润
需要注意的是,题目中要求不能同时参与多笔交易,即每次买入股票后,必须卖出,才能进行下一次买入股票的操作。
代码
def stock3(prices,k):
if len(prices)<2:
return 0
dp = [[[0]*(k+1),[0]*(k+1)] for _ in range(len(prices))]
#dp[i][0][j]表示第i天,第j次不持有股票的最大利润
#dp[i][1][j]表示第i天,第j次持有股票的最大利润
for j in range(1,k+1): #第l天持有股票,利润都为负数
dp[0][1][j] = -prices[0]
for i in range(1,len(prices)):
for j in range(1,k+1):
dp[i][0][j] = max(dp[i-1][0][j],dp[i-1][1][j]+prices[i]) #第i天,第j次持有股票只能由前一天的相同状态转化而成,或由前一天第j次持有股票的状态卖出股票转化而成
dp[i][1][j] = max(dp[i-1][1][j],dp[i-1][0][j-1]-prices[i]) #同上,但需要注意的是,前一天第j-1次不持有股票的状态,买入股票后,转换为当前状态。
print(dp)
return dp[len(prices)-1][0][k]