股票买卖问题-最多完成k次交易

问题
给定一个整数数组 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]
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值