123. 买卖股票的最佳时机 III

本题相当于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

你可能感兴趣的:(LeetCode,序列,动态规划)