LeetCode 309
考虑三种状态的转换,买入,卖出,cooldown,买入之后可以保持不动继续保持买入的状态,卖出之后进入cooldown,在cooldown也可以保持不变。cooldown之后可以买入。根据这个状态转换我们可以有这样的公式
buy【i】 = min(之前买的 buy【i-1】,cooldown之后买的 price【i】 - cooldown【i-1】)
sell【i】= price【i】-buy【i-1】(卖出股票)
cooldown【i】 = max(sell【i-1】卖出之后进入cooldown,cooldown【i-1】保持原来状态)
转为代码:
def maxProfit(self, prices: List[int]) -> int:
l = len(prices)
if l == 0: return 0
buy = [0]*l
sell = [0] * l
cooldown = [0]* l
buy[0] = prices[0]
for i in range(1, l):
buy[i] = min(buy[i-1], prices[i] - cooldown[i-1])
sell[i] = - buy[i-1] + prices[i]
cooldown[i] = max(sell[i-1], cooldown[i-1])
return max(sell[l-1], cooldown[l-1])