【Leetcode 188】买入股票的最佳时机Ⅳ

问题描述:

给定一个数组,它的第 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 122:买入股票的最佳时机Ⅱ”一样。 接下来创建2k的数组动态数组dp, 偶数时买入,奇数时卖出。

dp[0] = -prices[0];
dp[1] = max(dp[1], dp[0] - prices[1]);
dp[2] = max(dp[1], dp[1] + prices[2]);

以此类推 ... ...

参考实现过程

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        if(prices.size() == 0 || k<1)  return 0;
        int n = prices.size();
        int temp = 0;
        //判断是否无限进行交易
        if(k>n/2){
            for(int i=1; i<n; i++)
                temp += max(0, prices[i]-prices[i-1]);
            return temp;
        }
        vector<int>dp(2*k, INT_MIN);
        dp[0] = -prices[0];

        //sell: dp[j] = max(dp[j], dp[j-1] + prices[i])
        //buy:  dp[j] = max(dp[j], dp[j-1] - prices[i])
        for(int i=1; i<n; i++){
            dp[0] = max(dp[0], -prices[i]);
            for(int j=1; j<2*k; j++){
                if(j%2)
                    dp[j] = max(dp[j], dp[j-1] + prices[i]);
                else
                    dp[j] = max(dp[j], dp[j-1] - prices[i]);

            }
        }
        return max(0, dp.back());

    }
};

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值