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天可以获得的最大收益,第i天卖股票可以获得的最大收益。算法复杂度为。
2)将prices数组都遍历一遍之后,可以知道最后一项,即maxProfitWith[size-1]即为遍历一遍时的最大收益。
3)我们再逆序遍历一遍,计算当前i到最后一天的最大收益,,第i天买股票可以获得的最大收益。与之前的0至第i天的最大收益相加,可以得到总的最大收益finalProfit,比较之后,可以得到最终结果,算法复杂度也是。
4)最终的算法复杂度就是。
class Solution {
public:
int maxProfit(vector<int> &prices) {
if(prices.size()<=1)
return 0;
vector<int> sellProfit;
int lowest = prices[0];
int maxSell = 0;
sellProfit.push_back(0);
for(int i=1;i<prices.size();i++) {
int profit = prices[i]-lowest;
if(profit>maxSell)
maxSell = profit;
sellProfit.push_back(maxSell);
if(prices[i]<lowest)
lowest = prices[i];
}
int res = maxSell;
int highest = prices[prices.size()-1];
int maxBuy = 0;
for(int i=prices.size()-2;i>=0;i--) {
int profit = highest-prices[i];
if(profit>maxBuy)
maxBuy = profit;
if(res<maxBuy+sellProfit[i])
res = maxBuy + sellProfit[i];
if(prices[i]>highest)
highest = prices[i];
}
return res;
}
};