121. Best Time to Buy and Sell Stock
Say you have an array for which the ith element is the price of a given stock on day i.
If you were only permitted to complete at most one transaction (ie, buy one and sell one share of the stock), design an algorithm to find the maximum profit.
题目要求的是最多一次交易(包含一次买一次卖),可得的最大收益。最直观的想法就是,在第i天买入后,在第i+1到第n天的价格中找到一个比第i天价格大且最大的价格,这样做的话也能解决问题,只是时间复杂度为O(n^2),超时。换个思路,要求的是最大收益,那就意味着以尽量低的价格买入,最高的价格卖出。则我们需要在0~i天中寻找一个最小价格,并在i+1~n天中寻找一个最大价格,代码中则用buy来记录最小价格,sell来记录最大价格,buy和sell可在同一次循环中更新,即到第i天,价格依旧在递减,此时是售出收益为0,也是合理的,但收益不能为负数。若在i天之前价格序列不是持续递减的,则必然可以找到一个最小价格。同理,若后面部分也出现这种情况,前面的收益若是最佳,后面就不会影响到前面,若后面部分更佳,则根据min和max的选择,buy和sell则会被替换为最小购入价格和最大售出价格。具体代码如下:
#define min(m, n) (m < n ? m : n)
#define max(m, n) (m > n ? m : n)
int maxProfit(int* prices, int pricesSize) {
int buy;
int sell;
int i;
buy = prices[0];
sell = 0;
for (i = 1; i < pricesSize; ++i) {
buy = min(buy, prices[i]);
sell = max(sell, prices[i] - buy);
}
return sell;
}