题目描述
给定一个数组,它的第 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。
思路分析
- 这个我觉得还是很巧妙的
- prices长度为0时,无法购买股票吧,返回0
- 如果交易序列存在,初始化begin和result,如果赚不到钱(卖出价格大于买入价格),那么还是返回0
- 考虑赚得到钱的情况,后续序列形态如下:
[1,2,4,6,3] 很自然的,begin不用更新,result更新策略为旧result和遍历的p-begin中较大的那一项
[3,2,6,1,2] 这种情况下,begin存在更新,需要考虑在最大收益为6-3的情况下后续交易日能否出现最大值的情况。result记录了当前情况下的最大值
[2,3,6,1,9] 后续交易日有最大值,更新的第4日交易价格低于第1日,所以begin为1的最大值肯定大于begin为2的最大值。
代码实现
class Solution:
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
if len(prices) == 0:
return 0
begin = prices[0]
result = 0
for p in prices:
result = max(result,p-begin) #理解滴关键
begin = min(begin,p) #理解滴关键
return result