121. 买卖股票的最佳时机(Python)

题目

难度:★☆☆☆☆
类型:数组

给定一个数组,它的第 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。

解答

股票利润是卖出时间t2时的价格与买入时间t1时的价格之差,这里有t2>t1存在,为了获得最大利润,卖出时的价格与买入时价格的差值最大,我们可以考虑使用暴力求解法寻找所有的买入、卖出价格对,计算利润并更新利润最大值,但是计算复杂度较高,这里不再赘述。

我们使用一次遍历方法,用变量(cur_max_profit)和(cur_min)表示遍历到当前位置为止出现过的最大利润和当前的最小价格,在每次遍历时,如果遇到更小的股票价格,则更新cur_min,当前利润的计算方式是当前价格与cur_min之间的差值,如果遇到更大当前利润,则更新变量cur_max_profit,最终返回cur_max_profit即可。

class Solution:
    def maxProfit(self, prices):

        if not prices:                              # 输入为空
            return 0                                # 返回零

        cur_max_profit, cur_min = 0, prices[0]      # 初始化记录中最高利润和记录中最小价格
        for i in range(len(prices)-1):              # 进行迭代
            if prices[i] < cur_min:                 # 如果当前价格小于记录中最小价格
                cur_min = prices[i]                 # 更新记录中最小价格
            cur_profit = prices[i+1] - cur_min      # 当前利润=当前价格-记录中最小价格
            if cur_profit > cur_max_profit:         # 如果当前利润大于记录中最高利润
                cur_max_profit = cur_profit         # 更新记录中最高利润

        return cur_max_profit                       # 返回最高利润

如有疑问或建议,欢迎评论区留言~

你可能感兴趣的:(121. 买卖股票的最佳时机(Python))