【Leetcode】Best Time to Buy and Sell Stock III

Say you have an array for which the ith element is the price of a given stock on day i.

Design an algorithm to find the maximum profit. You may complete at most two transactions.

Note:
You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).

题意:获取股票最大收益,最多可交易两次,且必须完成一桩交易之后才能进行另一桩交易。当然,可以第i天卖出后又当天买回。

因为最多两次交易,很容易联想到将数组分为两部分,对两部分数组分别求最大值,然后将两个最大值相加,经多次比较,得到总的最大值。但是这会出现一个问题,时间复杂度太大,为。经测试,这种方式会导致Time Limited Exceed的错误。

那么现在需要考虑一下有没有可以改进的方法,答案是有的,可以利用一维的动态规划来做,因为我们可以观察到,之前的方案之所以复杂度太大是因为有很多重复的计算。而动态规划可以采取的方式是:

1)新开一个数组maxProfitWith,这个数组用来保存第i天可以获得的最大收益。算法复杂度为

2)将prices数组都遍历一遍之后,可以知道最后一项,即maxProfitWith[size-1]即为遍历一遍时的最大收益。

3)我们再逆序遍历一遍,计算当前i到最后一天的最大收益,与之前的0至第i天的最大收益相加,可以得到总的最大收益finalProfit,比较之后,可以得到最终结果,算法复杂度也是

4)最终的算法复杂度就是

class Solution {
public:
    int maxProfit(vector<int> &prices) {
        if(prices.empty())
            return 0;
        int size=prices.size();
        int lowest=prices[0];
        vector<int> maxProfitWith;
        maxProfitWith.push_back(0);
        int maxProfit=0;
        for(int i=1;i<size;i++)
        {
            int profit=prices[i]-lowest;
            if(profit>maxProfit)
                maxProfit=profit;
            maxProfitWith.push_back(maxProfit);
            if(prices[i]<lowest)
                lowest=prices[i];
        }
        int highest=prices[size-1];
        int finalProfit=maxProfitWith[size-1];
        maxProfit=0;
        for(int i=size-2;i>=0;i--)
        {
            int profit=highest-prices[i];
            if(profit>maxProfit)
                maxProfit=profit;
            int maxSum=maxProfitWith[i]+maxProfit;
            if(maxSum>finalProfit)
                finalProfit=maxSum;
            if(prices[i]>highest)
                highest=prices[i];
        }
        return finalProfit;
    }
};


  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值