【LeetCode】121. Best Time to Buy and Sell Stock 解题报告(Python)

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接: https://blog.csdn.net/ttinch/article/details/102605921

【LeetCode】121. Best Time to Buy and Sell Stock 解题报告(Python)

题目地址:https://leetcode.com/problems/best-time-to-buy-and-sell-stock/

题目描述

Say you have an array for which the ith element is the price of a given stock on day i.

If you were only permitted to complete at most one transaction (i.e., buy one and sell one share of the stock), design an algorithm to find the maximum profit.

Note that you cannot sell a stock before you buy one.

Example 1:

Input: [7,1,5,3,6,4]
Output: 5
Explanation: Buy on day 2 (price = 1) and sell on day 5 (price = 6), profit = 6-1 = 5.
             Not 7-1 = 6, as selling price needs to be larger than buying price.

Example 2:

Input: [7,6,4,3,1]
Output: 0
Explanation: In this case, no transaction is done, i.e. max profit = 0.

解法

只需要遍历一次数组,用一个变量记录遍历过数中的最小值,然后每次计算当前值和这个最小值之间的差值最为利润,然后每次选较大的利润来更新。当遍历完成后当前利润即为所求,代码如下:

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        ans = 0
        min_p = 10000
        for i in range(len(prices)):
            ans = max(ans, prices[i]-min_p)
            min_p = min(min_p, prices[i])
        return ans

猜你喜欢

转载自blog.csdn.net/ttinch/article/details/102605921