题目描述
风口之下,猪都能飞。当今中国股市牛市,真可谓“错过等七年”。 给你一个回顾历史的机会,已知一支股票连续n天的价格走势,以长度为n的整数数组表示,数组中第i个元素(prices[i])代表该股票第i天的股价。 假设你一开始没有股票,但有至多两次买入1股而后卖出1股的机会,并且买入前一定要先保证手上没有股票。若两次交易机会都放弃,收益为0。 设计算法,计算你能获得的最大收益。 输入数值范围:2<=n<=100,0<=prices[i]<=100
示例1
输入
3,8,5,1,7,8
输出
12
分析:
因为要求两次交易后的最大收益,所以先正序扫一遍记录第k天能获得的最大收益,然后逆序扫表示第k天之后的区间能获得的最大收益,然后相加求最大值
利用动态规划的思想。
首先从左往右扫描,先计算出子序列[0,...,i]中的最大利润,用一个left数组保存下来,时间是O(n)。第二步是从右往左逆向扫描,计算子序列[i,...,n-1]上的最大利润,
用right数组保存下来,s时间是O(n),同时结合left和right数组的结果就能计算最终的最大利润了,这一步也是O(n)。 所以最后算法的复杂度就是O(n)的。
code:
class Solution {
public:
/**
* 计算你能获得的最大收益
*
* @param prices Prices[i]即第i天的股价
* @return 整型
*/
int calculateMax(vector<int> prices) {
int len = prices.size();
vector<int> first(101);
vector<int> second(101);
int temp1=prices[0];
first[0]=0;
for(int i=1;i<len;i++){
temp1=min(temp1,prices[i]);
first[i]=max(first[i-1],prices[i]-temp1);
}
int temp2=prices[len-1];
second[len-1]=0;
for(int i=len-2;i>=0;i--){
temp2=max(temp2,prices[i]);
second[i]=max(second[i+1],temp2-prices[i]);
}
int result=0;
for(int i=0;i<len;i++){
result=max(result,first[i]+second[i]);
}
return result;
}
};