数据结构
网络
关系型数据库管理系统
操作系统
Java
MS Excel
iOS
HTML
CSS
Android
Python
C语言编程
C++
C#
MongoDB
MySQL
Javascript
PHP
物理学
化学
生物学
数学
英语
经济学
心理学
社会学
服装设计
法律研究Python程序：在最多买卖两次股票的情况下找到最大利润
假设我们有一个名为prices的数字列表，它按时间顺序表示公司的股票价格，我们必须找到在最多买卖两次股票的情况下可以获得的最大利润。我们必须先买入然后卖出。
因此，如果输入类似于 prices = [2, 6, 3, 4, 2, 9]，则输出将为 11，因为我们可以在价格 2 买入，然后在 6 卖出，再次在 2 买入，并在 9 卖出。
为了解决这个问题，我们将遵循以下步骤 -
- first_buy := -inf，first_sell := -inf
- second_buy := -inf，second_sell := -inf
- 对于 prices 中的每个 px，执行以下操作
- first_buy := first_buy 和 -px 的最大值
- first_sell := first_sell 和 (first_buy + px) 的最大值
- second_buy := second_buy 和 (first_sell - px) 的最大值
- second_sell := second_sell 和 (second_buy + px) 的最大值
- 返回 0、first_sell 和 second_sell 的最大值
让我们看看以下实现以更好地理解 -
示例
class Solution:
def solve(self, prices):
first_buy = first_sell = float("-inf")
second_buy = second_sell = float("-inf")
for px in prices:
first_buy = max(first_buy, -px)
first_sell = max(first_sell, first_buy + px)
second_buy = max(second_buy, first_sell - px)
second_sell = max(second_sell, second_buy + px)
return max(0, first_sell, second_sell)
ob = Solution()
prices = [2, 6, 3, 4, 2, 9]
print(ob.solve(prices))输入
[2, 6, 3, 4, 2, 9]
输出
11
更新于： 2020年12月2日
223 次浏览
广告