[leetcode 121] 买卖股票的最佳时机
[leetcode 121] 买卖股票的最佳时机

Problem

121. 买卖股票的最佳时机

给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。

如果你最多只允许完成一笔交易(即买入和卖出一支股票),设计一个算法来计算你所能获取的最大利润。

注意你不能在买入股票前卖出股票。

示例 1:

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

示例 2:

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

Solution

时间复杂度为O(N^2)的两遍遍历的方法在此就不做赘述了,下面着重介绍这种时间复杂度为O(N)的算法。

其实这种O(N)的方法也是比较简单,在一遍遍历的时候,需要两个判断:

  1. 该节点是不是最小值(买股票)
  2. 该节点值 - 最小值 是不是最大利润(卖股票)

因为是从头至尾遍历,还避免了2出现在1之前的情况。

func maxProfit(prices []int) int {

	length := len(prices)

	if length < 2 {
		return 0
	}

	minPrice := prices[0]
	maxProfit := 0

	for i := 1; i < length; i++ {
		if prices[i] < minPrice {
			minPrice = prices[i]
		}

		if maxProfit < prices[i]-minPrice {
			maxProfit = prices[i] - minPrice
		}

	}
	return maxProfit
}

If You Have Any Question, You Can Contact Me Through liam@blue7wings.com, @Blue7Wings, #Liam_Hsia