本题相当于T121和T122的进阶版
相比于前两次的只有一次交易和不限交易次数,本题仅限小于等于两次交易
虽然有前两题做铺垫,但我还是没有思路
自己想的是一些暴力方法,复杂度基本都是O(n^2)左右
看了一下题解,有点像动态规划,基本上都是以下思路:
设定四个变量:
fst_buy第一次买入可以获得的最大利润
fst_sell第一次卖出可以获得的最大利润
sec_buy第二次买入可以获得的最大利润
sec_sell第二次卖出可以获得的最大利润
由于四个时间节点都是递进的,是有严格先后顺序的
所以前值会影响后值,后值不会影响前值,且每个状态满足无后效性
代码如下(处理是相当于可以当天买卖再买卖,结果是一样的):
from typing import List
class Solution:
def maxProfit(self, prices: List[int]) -> int:
if len(prices) == 0:
return 0
fst_buy = - max(prices) - 100
fst_sell = 0
sec_buy = - max(prices) - 100
sec_sell = 0
for p in prices:
fst_buy = max(fst_buy, -p)
fst_sell = max(fst_sell, fst_buy + p)
sec_buy = max(sec_buy, fst_sell - p)
sec_sell = max(sec_sell, sec_buy + p)
return sec_sell