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).
思路:
用两个数组保存中间变量,profit1保存在第i天之前交易一次所得的最大利润,(正序遍历,记录minPrice)
profit2保存在第i天之后交易一次所得的最大利润,(逆序遍历,记录maxPrice)
在将两个数组对应位的值相加的最大值即交易两次所得的最大利润。
代码:
int maxProfit(vector<int> &prices) {
int len=(int)prices.size();
if(len<=1)
{
return 0;
}
int *profit1=new int[len];
int *profit2=new int[len];
for(int i=0; i<len; ++i)
{
profit1[i]=0;
profit2[i]=0;
}
int minPrice=prices[0];
int maxPrice=prices[len-1];
for(int i=1; i<len; ++i)
{
minPrice=min(minPrice,prices[i]);
profit1[i]=max(profit1[i-1],(prices[i]-minPrice));
}
for(int i=len-2; i>=0; --i)
{
maxPrice=max(maxPrice,prices[i]);
profit2[i]=max(profit2[i+1],(maxPrice-prices[i]));
}
int maxPro=0;
for(int i=0; i<len; ++i)
{
if(profit1[i]+profit2[i]>maxPro)
{
maxPro=profit1[i]+profit2[i];
}
}
delete []profit1;
delete []profit2;
return maxPro;
}