题目描述
给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
如果你最多只允许完成一笔交易(即买入和卖出一支股票一次),设计一个算法来计算你所能获取的最大利润。
注意你不能在买入股票前卖出股票。
示例 1:
输入: [7,1,5,3,6,4]
输出: 5
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格。
示例 2:
输入: [7,6,4,3,1]
输出: 0
解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。
我的解法:两层迭代器循环的暴力法运行超时
class Solution {
public:
int maxProfit(vector<int>& prices) {
if(prices.empty()) return 0;
int max_profit = 0;
for(vector<int>::iterator iter = prices.begin(); iter != prices.end() - 1; iter++)
{
for(vector<int>::iterator iter1 = iter+1; iter1 != prices.end(); iter1++)
{
int profit = *iter1 - *iter;
if(profit > max_profit) max_profit = profit;
}
}
return max_profit;
}
};
正确解法:一次遍历,记录每次循环的最小值
只执行一次循环,在每一次循环都拿当前值和之前得到的最小值比较,如果当前值小于最小值,则当前值成为新的最小值,若当前值大于最小值,则进行相减计算利润,便利每一次循环,找出利润最高的。
class Solution {
public:
int maxProfit(vector<int>& prices) {
if(prices.empty()) return 0;
int min_price = prices[0];
int max_profit = 0;
for(int i = 1; i < prices.size(); i++)
{
if(prices[i] > min_price)
{
int profit = prices[i] - min_price;
if(profit > max_profit) max_profit = profit;
}
else
{
min_price = prices[i];
}
}
return max_profit;
}
};