LeetCode-121. 买卖股票的最佳时机

贪心算法#

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

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

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

示例 1：

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

示例 2：

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

思路

ai润色后的思路，就是之前用切片只关注栈顶元素，有点浪费空间(这里我有点理解最小栈不止访问上一个元素)，这里只需要一个变量即可

1
func maxProfit(prices []int) int {
2
    minPrice := prices[0]
3
    profit := 0
4

5
    for _, price := range prices {
6
        minPrice = min(minPrice, price)
7
        profit = max(profit, price - minPrice)
8
    }
9

10
    return profit
11
}

手撕思路

可能是之前刚做完最小栈的题目，这里我自然而然地想到了，就是用一个切片储存当前的最小值，profit = price - minVal即可

1
func maxProfit(prices []int) int {
2
    var stack []int
3
    profit := 0
4

5
    for _, v := range prices {
6
        minVal := v
7
        if len(stack) > 0 {
8
            if v > stack[len(stack)-1] {
9
                minVal = stack[len(stack)-1]
10
            }
11
        }
12

13
        profit = max(profit, v - minVal)
14
        stack = append(stack, minVal)
15
    }
16

17
    return profit
18
}