Difficulty:Hard
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.
解题思路(时间复杂度O(n)):
在这道题目中的解决方案需要采用分治的策略,并且动态规划每一步的状态,而状态传递方程在这个题目中实际就是dis[0][i]=ans;
在题目中计算中,可以将整个问题看做划分为2部分,以i为界,在i之前的序列中找到一个最大的profit,再在i之后找到一个最大地profit,并且将这些值记录起来,而记录这些位置的值所使用的方法其实是动态规划分治策略的简化版,只使用一个二维数组,并且大小为2*n的大小。
而计算前半部分的收益与后半部分的收益其实就是解决123. Best Time to Buy and Sell Stock I的时候使用的方法,在这里不赘述了。
首先进行一次遍历,并且对头尾分别作出处理,使用i来不断更新dis[0][i]的值,也就是记录前半部分的最大值。而使用n-i-1不断更新dis[1][n-1-i]的值,也就是后半部分的最大值。所以相当于经过一次遍历之后,二维数组中已经存储了对于每一个位置i的前半部分的最大收益以及后半部分的最大收益
最后对i进行一次遍历,计算出对应的每一个i位置的总收益temp=dis[0][i]+dis[1][i];
而有人可能会考虑到如下问题,题目中说了最多只交易两次,但是不一定必须要两次,可能交易一次也可以获得最大值。
那么交易一次就得到最大的收益其实对应的状况就是在前半部分最终在i位置卖出,而在后半部分i位置卖出,这其实就等于在i位置既没有卖出也没有买入,因为这种状况是不允许的,所以就是等一进行的一次交易。而假如在前半部分或者后半部分没有收益,那么其实也就是进行的一次交易,所以对这个问题的考虑是没有必要进行特殊化的。
代码如下:
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n=prices.size();
int dis[2][n]={0};
int min=10000000;
int ans=0;
int ans1=0;
int max=-10000000;
for(int i=0;i<n;i++)
{
if(min>prices[i]) min=prices[i];
else if(prices[i]-min>ans)
{
ans=prices[i]-min;
}
dis[0][i]=ans;
if(max<prices[n-i-1]) max=prices[n-i-1];
else if(prices[n-1-i]-max<ans1)
{
ans1=prices[n-1-i]-max;
}
dis[1][n-1-i]=-ans1;
}
int temp=0;
for(int i=0;i<n-1;i++)
{
if(dis[0][i]+dis[1][i]>temp) temp=dis[0][i]+dis[1][i];
}
return temp;
}
};