题目描述:给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
解题思路:动态规划,每天交易结束时可能有两种情况,一是手上有股票二是手上无股票,然后在这个基础上加上交易次数,当前次数的交易次数只和当前和上一次交易次数有关,然后进行关系转移,代码如下:
class Solution:
def maxProfit(self, prices: List[int]) -> int:
n = len(prices)
buy, sell = [-prices[0], -prices[0]], [0, 0] #buy 是当天持有 sell是当天不持有
for i in range(1, n):
buy[0] = max(-prices[i], buy[0])
sell[0] = max(buy[0]+prices[i], sell[0])
buy[1] = max(buy[1], sell[0]-prices[i])
sell[1] = max(buy[1]+prices[i], sell[1])
return sell[1]