leetcode - Best Time to Buy and Sell Stock - 买卖股票的最佳时机

2024年6月10日 94次阅读

端午节没事，在家刷刷LeetCode ~，顺便学习golang

题目地址

题解

问题就是找到数组中最大值同最小值的差，且最大值在最小值后面。开始的想法是遍历数组两次，计算从当前值开始的最大利润。这样时间复杂度就是 O(n2) O ( n 2 ) ，也通过了

func maxProfit(prices []int) int {
    arrLen := len(prices)
    maxProfit := 0
    for i := 0; i < arrLen; i++ {
        for j := i + 1; j < arrLen; j++ {
            price1 := prices[i]
            price2 := prices[j]
            profit := price2 - price1
            if profit > maxProfit {
                maxProfit = profit
            }
        }
    }

    return maxProfit
}

网友给出了 O(n) O ( n ) 的解法，思路是遍历一次数组，记录目前为止的最小值，用当前值减最小值，和最大利润比较，我也实现了一遍：

func maxProfit(prices []int) int {
    if len(prices) == 0 {
        return 0
    }

    min := prices[0]
    res := 0
    for _, val := range prices {
        if val < min {
            min = val
        }
        diff := val - min
        if diff > res {
            res = diff
        }
    }

    return res
}

学到 go 的用法：

  • 不支持三元运算符，用if代替
  • for index, val := range arr range 遍历数组
点赞