日拱一卒（5）——leetcode学习记录：股票购买

一、任务

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

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

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

二、思路

1. 暴力求解（超时）

遍历价格中的元素，及该元素后最大的元素，计算利润，更新最大利润

问题：每次计算最大值都需要遍历一次，如果改为复用就能减少时间

2. 优化

遍历价格中的元素，以及该元素前的最小元素，计算利润，更新最大利润

三、解答

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        profit = 0
        min_val = prices[0]
        for i,item in enumerate(prices[1:]):
            if item < min_val:
                min_val = item
            else:
                new_profit = item - min_val
                if new_profit > profit:
                    profit = new_profit
        return profit