继续刷LeetCode,第121题,求最大股票收益,也是求数组中最大的差异,并且小的值在前,大的值在后。
分析:
设置两个变量,存放一大一小两个值,遍历整个数组,碰到比小值还小的,就更新较小值,计算下一个值与较小值的差,使这个差最大。这个求解过程就是不断更新较小值以及差值。
问题:
1、代码的优化;
附上代码:
class Solution {
public:
int maxProfit(vector<int>& prices) {
if(prices.empty())
return 0;
int buy=prices[0],sell;
int profit=0;
for(int i=1;i<prices.size();i++)
{
sell=prices[i];
if(sell<buy)
{
buy=sell;
continue;
}
else
{
if((sell-buy)>profit)
profit=sell-buy;
}
}
return profit;
}
};
附上优化后的代码:
class Solution {
public:
int maxProfit(vector<int>& prices) {
if(prices.empty())
return 0;
int buy=prices[0];
int profit=0;
for(int i=1;i<prices.size();i++)
{
buy=min(buy,prices[i]);
profit=max(profit,prices[i]-buy);
}
return profit;
}
};
利用数组相邻位置关系,最优的代码:
class Solution {
public:
int maxProfit(vector<int>& prices) {
int ret = 0, max = 0;
for(int i = 1; i < prices.size(); i++)
{
ret += prices[i] - prices[i-1];
if(ret < 0) ret = 0;
if(ret > max) max = ret;
}
return max;
}
};