python买卖股票_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 # 返回最高利润

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