描述
假设你有一个数组,其中第i个元素是某只股票在第i天的价格。
设计一个算法来求最大的利润。你最多可以进行两次交易。
注意:
你不能同时进行多个交易(即,你必须在再次购买之前出售之前买的股票)。
示例1
输入:
[1,4,2]
返回值:
3
示例2
输入:
[2,4,1]
返回值:
2
class Solution {
public:
int maxProfit(vector<int> &prices) {
//You may complete at most two transactions.(you must sell the stock before you buy again)
//方式一:爆搜 (630ms 8552k)
/*
int len=prices.size();
int maxPro=0;
for(int i=0;i<len;i++){ //以i作为两次交易的分界线
int max1=0,max2=0;
for(int j=0;j<=i;j++){
for(int k=j+1;k<=i;k++){
int val = prices[k]-prices[j];
if(val>max1){
max1=val;
}
}
}
for(int j=i;j<len;j++){
for(int k=j+1;k<len;k++){
int val = prices[k]-prices[j];
if(val>max2){
max2=val;
}
}
}
if(max1+max2>maxPro)maxPro = max1+max2;
}
return maxPro;
*/
//方式二:波峰波谷检测法,可以减小不必要的搜索空间,相比于第一种效率提升了不少 (180ms 8568k)
int len = prices.size();
if(len<=1)return 0;
vector<int> flag(len,0);
for(int i=1;i<len-1;i++){
if(prices[i]>prices[i-1]&&prices[i]>=prices[i+1])flag[i]=1; //标记为波峰(比较时注意相等情况)
if(prices[i]<prices[i-1]&&prices[i]<=prices[i+1])flag[i]=-1; //标记为波谷
}
if(prices[0]<=prices[1])flag[0]=-1;
if(prices[len-1]>prices[len-2])flag[len-1]=1;
//搜索
int maxPro=0;
for(int k=0;k<len;k++){
int max1=0,max2=0;
for(int i=0;i<=k;i++){
if(flag[i]==-1){
for(int j=i+1;j<=k;j++){
if(flag[j]==1){
int val = prices[j]-prices[i];
if(val>max1)max1 = val;
}
}
}
}
for(int i=k;i<len;i++){
if(flag[i]==-1){
for(int j=i+1;j<len;j++){
if(flag[j]==1){
int val = prices[j]-prices[i];
if(val>max2)max2 = val;
}
}
}
}
int val = max1+max2;
if(val>maxPro)maxPro = val;
}
return maxPro;
}
};