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
- 对于价格中的每个px,执行
- 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))
- 返回0、first_sell和second_sell中的最大值
让我们看下面的实现以更好地理解-
更多Python相关文章,请阅读:Python 教程
示例
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