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

题解:

代码:

class Solution {
    public int maxProfit(int[] prices) {
        // 核心思路在于计算每日卖出时的利润最大值
        int cost = Integer.MAX_VALUE;
        int profit = 0;
        for(int i = 0; i < prices.length; i++) {
            // 在遍历到当前元素时, cost 已经是遍历过的元素中的最小值了
            if(prices[i] - cost > profit) {
                profit = prices[i] - cost;
            } 
            cost = Math.min(cost, prices[i]); 
        }
        return profit;
    }
}

核心思路:

  • 我觉得核心思路是模拟真实买卖股票的流程,以时间为横轴,最大利润为纵轴,假设我在今天卖掉股票,那我的最大收益是多少?
  • 那有卖就有买,想要在今天卖掉股票并获得最大收益,那我是不是得在历史的价格最低点买入,这样我在今日卖出的收益才是最大的。
  • 想通了上述的流程,就可以写出代码了,以 cost 代表历史价格最低点,遍历计算每天的利润最大值时,同时维护最低点买入价格

如果想要再精简一下代码:

class Solution {
    public int maxProfit(int[] prices) {
        // 核心思路在于计算每日卖出时的利润最大值
        int cost = Integer.MAX_VALUE;
        int profit = 0;
        for(int i = 0; i < prices.length; i++) {
            // 在遍历到当前元素时, cost 已经是遍历过的元素中的最小值了
            profit = Math.max(prices[i] - cost, profit);
            cost = Math.min(cost, prices[i]); 
        }
        return profit;
    }
}