编写一段python代码,实现这个功能:假设你有一个数组prices,长度为n,其中prices[i]是某只股票在第i天的价格,请根据这个价格数组,返回买卖股票能获得的最大收益 1. 你最多可以对该股票有两笔交易操作,一笔交易代表着一次买入与一次卖出,但是再次购买前必须卖出之前的股票 2. 如果不能获取收益,请返回0 3. 假设买入卖出均无手续费 数据范围:1 ≤n≤10^5,股票的价格满足 1 ≤val≤10^4 要求: 空间复杂度 O(n),时间复杂度 O(n) 进阶:空间复杂度 O(1),时间复杂度 O(n)
时间: 2023-02-09 09:57:10 浏览: 141
以下是第一种做法的 Python 代码:
```
def max_profit(prices):
n = len(prices)
if n < 2:
return 0
# 分别计算前i天的最大收益,以及前i天的最小价格
max_profits = [0] * n
min_prices = [0] * n
min_prices[0] = prices[0]
for i in range(1, n):
min_prices[i] = min(min_prices[i - 1], prices[i])
max_profits[i] = max(max_profits[i - 1], prices[i] - min_prices[i])
# 分别计算后i天的最大收益,以及后i天的最大价格
reverse_max_profits = [0] * n
max_prices = [0] * n
max_prices[-1] = prices[-1]
for i in range(n - 2, -1, -1):
max_prices[i] = max(max_prices[i + 1], prices[i])
reverse_max_profits[i] = max(reverse_max_profits[i + 1], max_prices[i] - prices[i])
# 最大收益就是两次交易的最大收益之和
max_profit = 0
for i in range(n):
max_profit = max(max_profit, max_profits[i] + reverse_max_profits[i])
return max_profit
```
以下是第二种做法的 Python 代码:
```
def max_profit(prices):
n = len(prices)
if n < 2:
return 0
# 分别计算前i天的最大收益,以及前i天的最小价格
max_profit = 0
min_price = float('inf')
for i in range(n):
min_price = min(min_price, prices[i])
max_profit = max(max_profit, prices[i] - min_price)
# 分别计算后i天的最大收益,以及后i天的最大价格
second_max_profit = 0
max_price = float('-inf')
for i in range(n - 1, -1, -1):
max_price = max(max_price, prices[i])
second_max_profit = max(second_max_profit, max_price - prices[i])
# 最大
阅读全文