买卖股票的最佳时机IV

题目(from leetcode)
给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。

注意: 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。

示例 1:

输入: [2,4,1], k = 2
输出: 2
解释: 在第 1 天 (股票价格 = 2) 的时候买入,在第 2 天 (股票价格 = 4) 的时候卖出,这笔交易所能获得利润 = 4-2 = 2 。
示例 2:

输入: [3,2,6,5,0,3], k = 2
输出: 7
解释: 在第 2 天 (股票价格 = 2) 的时候买入,在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。
     随后,在第 5 天 (股票价格 = 0) 的时候买入,在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-iv
解法一(from leetcode-author yushuo-guan)
class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        int N = prices.size();
        if(N == 0) return 0;
        vector<int> dp(N, 0);
        //add-start:此处需要添加k很大的处理
        //add-end
        while(k-- > 0)
        {
            for(int i = 1; i < N; ++i)
                dp[i] = max(dp[i - 1] - prices[i - 1] + prices[i], dp[i]);
            for(int i = 1; i < N; ++i)
                dp[i] = max(dp[i], dp[i - 1]);
        }
        return dp[N - 1];
    }
};
/* 作者解释
while(k-- > 0)
{
    //第1个for循环中,dp[i]代表第i天完成前k次交易所能获得的最大价值。
    //第2个for循环中,dp[i]代表前i天完成前k次交易所能获得的最大价值。
}

第一个for循环中,
dp[i]的值有两种可能:
第一,第i天完成第k笔买和卖,第k笔收益为0。
第二,第i天完成第k笔的卖出,之前的一些天完成其他交易。满足这种要求的所有情况,都能和第i-1天完成第k笔交易的情况建立一对一的关系,只是第k笔交易的卖出价格由prices[i - 1]变为了prices[i]。所以dp[i] = dp[i - 1] - prices[i - 1] + prices[i]
*/
// 这部分是对k很大情况(e.g k=1e8)的处理,需添加到上面代码段里的add部分
if(k >= N / 2)
{
    ret = 0;
    for(int i = 1; i < N; ++i)
    {
        if(prices[i] > prices[i - 1])
            ret += prices[i] - prices[i - 1];
    }
    return ret;
}

作者:yushuo-guan
解法一解释
  • 本题是动态规划(dp)部分的,有典型的dp解法;上述解法一比较有意思,不太像dp解法,作者的解释没看太明白,所以我自己添油加醋一番
  • 以官方例子解释:
    • 输入: [3,2,6,5,0,3], k = 2 输出: 7
    • 参考下图在这里插入图片描述
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值