leetcode刷题--121.买卖股票的最佳时机

180 阅读2分钟

1. 题目描述

给定一个数组 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

2. 解题思路

  • 其实这题如果简单看,好像是找出数组中的最小和最大值,返回最大最小值的差即可。
  • 但是示例2中,其实已经给出了明确的反例,如果最大值出现的时间在最小值出现之前,比如第一天出现最大值7,但是股票一路下跌,到了第5天,股票跌到了1,那么此时交易肯定是亏本的,所以这个简单的思路是错误的。
  • 延续这个思路继续思考,其实在第N天能不能交易,需要看前N-1天中的最小值,最小值是不是大于第N天,如果大于则不能交易,否则可以交易;
  • 如果可以交易,那么就看下差值是不是最大的即可;
  • 为了方便实现,就用一个数组来保存前N-1天的最小值即可。

3. 编码实现

func maxProfit(prices []int) int {
	dpList := make([]int, len(prices)) // 用一个数组来保存前N-1天的最小值
	for idx, val  := range prices {
		if idx == 0 {
			dpList[idx] = val
		}else {
			dpList[idx] = dpList[idx-1]
			if dpList[idx-1] > val { //如果当前股票价格前idx-1 天的最小值还要小
				dpList[idx] = val //使用当前值作为最小值
			}
		}
	}
	retMax := 0
	for idx, val := range prices { //最后再遍历下,看下差值最大的返回即可
		if val - dpList[idx] > retMax {
			retMax = val-dpList[idx] 
		}
	}
	return retMax
}