动态规划思路:定义数组left[n],left[i]表示股票在当天能获得的最大收益,状态转移方程则可以表示为:left[i]=prices[i]-min>prices[i]?prices[i]-min:prices[i-1].
对于第二次买入卖出定义数组right[n],同理right[i]表示当天能获得最大收益。不过left数组是算出当前为止的最低价格并算出当天为止的最大差价。right数组是倒序算出最高价格并记录最大差价
max是倒序到当天的最高值,过程同上
总结一下,这道题有四个动态变化量,分别为第一次买入卖出的最小值min和最大差价left[i],min为直到当天股市的最低值,初始化为prices[0],最大利润left[i]初始化为0。如果当天卖出,(不考虑当天买当天卖的情况,遍历从第二天开始),则收益为当天价格prices[i]-min,如果大于到目前为止最大利润left[i-1],则更新left[i-1];否则最大利润继续为left[i-1],和前一天保持一致;
另外两个动态变化量为max和right[i],同理
至此我们就可以写出如下代码,采用两个数组双向遍历,避免重叠问题
class Solution { public: /** */ //以卖股票当日为分界点,另外注意股票可以当天卖了再买 int calculateMax(vector<int> prices) { int n=prices.size(); //下两个变量代表最大的差价 vector<int>left(n,0);//第一次买入卖出最大差价 vector<int>right(n,0);//第二次买入卖出最大差价。为了防止两次重叠问题,第二次从右向左遍历 int max=0,min=0; min=prices[0]; max=prices[n-1]; for(int i=1;i<n;i++)//从第二天就算与第一天的差价开始 { if(prices[i]<min) min=prices[i]; left[i]=prices[i]-min>left[i-1]? prices[i]-min:left[i-1];//当前的价格和目前为止最低价相减与上一日为止最大差价比较 } for(int j=n-2;j>=0;j--) { if(prices[j]>max) max=prices[j]; right[j]=max-prices[j]>right[j+1]?max-prices[j]:right[j+1]; } max=0; for(int i=0;i<n;i++) if(left[i]+right[i]>max) max=left[i]+right[i]; return max; } };
运行时间:3ms
占用内存:496k
同步到: 进入下一题 分享解题思路运行时间:3ms
占用内存:496k
同步到: 进入下一题 分享解题思路