题目描述
已知一个数组,第i个元素表示第i天股票的价格,你只能进行一次交易(买卖各一次),设计算法找出最大收益
测试样例
Input: [7, 1, 5, 3, 6, 4]
Output: 5
最大收益 = 6-1 = 5 (不是7-1 = 6,因为先买后卖,7买,1买亏了6)
Input: [7, 6, 4, 3, 1]
Output: 0
最大收益为0
详细分析
初看非常简单,遍历数组,每次选择一个元素,找到这个元素后面的数组的最大值,计算差值,和当前最大收益比较即可,就像这样:
[7,1,5,3,6,4] 当前7,后面最大6,收益-1
[7,1,5,3,6,4] 当前1,后面最大6,收益5
[7,1,5,3,6,4] 当前5,后面最大6,收益1
如此继续找到即可。
不过这种方法会超时,稍微改变一下,现在不止保持最大收益,还保存最低价格,如果当天价格更低,就刷新最小价格(买),同时如果当天价格减去最小价格的收益最大,就刷新最大价格(卖),过程如下:
[7,1,5,3,6,4] minPrice=INT_MAX,maxProfit=0
=> minPrice=7,maxProfit=0
[7,1,5,3,6,4] minPrice=7,maxProfit=0
=> minPrice=1,maxProfit=0
[7,1,5,3,6,4] minPrice=1,maxProfit=0
=> minPrice=1,maxProfit=4
[7,1,5,3,6,4] minPrice=1,maxProfit=4
=> minPrice=1,maxProfit=4
[7,1,5,3,6,4] minPrice=1,maxProfit=4
=> minPrice=1,maxProfit=5
[7,1,5,3,6,4] minPrice=1,maxProfit=5
=> minPrice=1,maxProfit=5
算法实现
- 方法1:Time Limit Exceeded
class Solution {
public:
int maxProfit(vector<int>& prices) {
int profit = 0;
for(auto start=prices.begin();start!=prices.end();start++){
int buy = *start;
int sell = *std::max_element(start,prices.end());
if((sell - buy)>profit){
profit = sell-buy;
}
}
return profit;
}
};
- 方法2: Accepted
class Solution{
public:
int maxProfit(vector<int> &prices){
int profit = 0;
int minBuy = std::numeric_limits<int>::max();
for(int i=0;i<prices.size();i++){
//buy it if current price is less than minimum prices yet
if(prices[i]<minBuy){
minBuy = prices[i];
}
//sell it if current profit got optimally
if((prices[i]-minBuy)>profit){
profit = prices[i]-minBuy;
}
}
return profit;
}
};