这道题还是挺难的,属于我前面提到的,给个数组,线性时间找出个什么东西,虽然上面的两个买卖股票也是这类,不过相比之下稚嫩多了。有关至少至多的问题比较烦人,不好想,等再做一些题,可能会发现什么规律。这道题的情况还是比较少的,要么买卖了两次,要么一次。
买卖一次的情况,已经解决过了,现在分析买卖两次的情况。两次买卖之间是没有交叉的,即下一次买之前一定已经卖掉了。最容易想到,穷去分点,每个部分都按照买卖一次的方法做。好,恭喜,你已经走在超时的路上了。那么怎么办呢,有没有一种方法,在线性时间内,等我走到一个分点的时候,就能知道划分两次求出来的结果?
没错,辅助空间。先从头往后扫一遍,记录下以这个点为终点时的最大收入,然后从尾向头扫一遍,记录下以这个点为起点的最大收入。再对每个分点扫一遍,看看已他为起点终点算出来的交易收入是不是最优的。
思路是这样的,描述起来比较清晰,实现的时候还是有可以优化的地方的,例如你发现从尾向头扫描的结果根本不用一整个数组,只要保存好后一个位置的最优值就可以了,因为当前最优值只跟峰值与当前值以及过去最优相关。
代码如下,未优化空间:
class Solution {
public:
int maxProfit(vector<int> &prices) {
int len = prices.size();
if(len <= 1)
return 0;
vector<int> phistory(len);
vector<int> pfuture(len);
int valley = prices[0], peak = prices[len-1];
for(int i=1;i<len;i++){
valley = min(valley, prices[i]);
phistory[i] = max(phistory[i-1], prices[i]-valley);
}
int res = 0;
for(int i=len-2;i>=0;i--){
peak = max(peak, prices[i]);
pfuture[i] = max(pfuture[i+1], peak-prices[i]);
res = max(res, pfuture[i]+phistory[i]);
}
return res;
}
};leetcode第一刷_Best Time to Buy and Sell Stock III,布布扣,bubuko.com
leetcode第一刷_Best Time to Buy and Sell Stock III
原文地址:http://blog.csdn.net/u012792219/article/details/25186747