Leetcode 121. 买卖股票的最佳时机

本题链接

题目

给定一个数组 prices ，它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。

你只能选择 某一天 买入这只股票，并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。

返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润，返回 0 。

示例 1：

输入：[7,1,5,3,6,4]
输出：5
解释：在第 2 天（股票价格 = 1）的时候买入，在第 5 天（股票价格 = 6）的时候卖出，最大利润 = 6-1 = 5 。
     注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格；同时，你不能在买入前卖出股票。

示例 2：

输入：prices = [7,6,4,3,1]
输出：0
解释：在这种情况下, 没有交易完成, 所以最大利润为 0。

提示：

1 <= prices.length <= 105
0 <= prices[i] <= 104

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

思路

假设买入了第一天的股票，其价格为 buyIn，之后遍历数组，执行这样的策略：

• 如果在新的一天发现价格比手上的 buyIn 要低，则把 buyIn 换成更低的股票（这样后面遇到更高的价格，卖掉的话肯定比之前的 buyIn 更赚）
• 如果在新的一天发现价格比手上的 buyIn 要高，就尝试着卖掉，并更新答案（但是 buyIn 仍然捏在手里，说不定后面有更高的价格）

时间复杂度：$O(n)$
空间复杂度：$O(1)$

C++ 代码

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if (prices.size() < 2)
            return 0;
        int buyIn = prices[0], ans = 0;
        for (int i = 1; i < prices.size(); ++i) {
            if (prices[i] < buyIn)
                buyIn = prices[i];
            else
                ans = max(ans, prices[i] - buyIn);
        }
        return ans;
    }
};