题目描述
风口之下,猪都能飞。当今中国股市牛市,真可谓“错过等七年”。 给你一个回顾历史的机会,已知一支股票连续n天的价格走势,以长度为n的整数数组表示,数组中第i个元素(prices[i])代表该股票第i天的股价。 假设你一开始没有股票,但有至多两次买入1股而后卖出1股的机会,并且买入前一定要先保证手上没有股票。若两次交易机会都放弃,收益为0。 设计算法,计算你能获得的最大收益。 输入数值范围:2<=n<=100,0<=prices[i]<=100
输入例子:
3,8,5,1,7,8
输出例子:
12
对于该问题,关键点在于认识到,做多只能进行两次交易(买入-卖出算一次)
那么,在一共n天里,前i天(第一天到第i天)完成一次交易,后面n-i天(第i+1天到第n天)完成第二次交易
两次交易利润和最大。
那么问题就分解成了一个动态优化问题:
首先求得前i天完成一次交易的最大利润,存储在left[i]中
再求得后n-i天完成交易的最大利润,存储在right[i+1]中
最后,根据left[i]和right[i+1]求解最大利润
class Solution {
public:
/**
* 计算你能获得的最大收益
*
* @param prices Prices[i]即第i天的股价
* @return 整型
*/
int calculateMax(vector<int> prices) {
int days = prices.size();
int* left = new int[days];
int* right = new int[days];
int min = INT_MAX;
int max_lirun = 0;
for(int i =0; i<days; i++){
if( prices[i] <min)
{ left[i] = max_lirun;
min = prices[i];
}
else
{
if( prices[i] - min > max_lirun )
{ max_lirun = prices[i] - min;}
left[i] = max_lirun;
}
}
int max = INT_MIN;
max_lirun = 0;
for(int i =days-1; i>=0 ; i--){
if( prices[i] > max )
{ right[i] = max_lirun;
max = prices[i];
}
else
{
if( max - prices[i] > max_lirun )
{ max_lirun = max - prices[i] ;}
right[i] = max_lirun;
}
}
max_lirun = left[days-1]>right[0]?left[days-1]:right[0];
for(int i = 0; i < days-1; i++){
if(left[i]+right[i+1] > max_lirun)
max_lirun = left[i]+right[i+1];
}
delete[] left;
delete[] right;
return max_lirun;
}
};