今天的还是buy&sell专题
714. Best Time to Buy and Sell Stock with Transaction Fee
这道题和昨天的题是差不多的,区别就是在hold递推公式上要加入-fee
class Solution:
def maxProfit(self, prices: List[int], fee: int) -> int:
dp_buy=[0 for _ in range(len(prices))]
dp_hold=[0 for _ in range(len(prices))]
dp_buy[0]=-prices[0]
for i in range(1,len(prices)):
dp_buy[i],dp_hold[i]=max(dp_buy[i-1],dp_hold[i-1]-prices[i]),max(dp_hold[i-1],dp_buy[i-1]+prices[i]-fee)
return dp_hold[-1]
309. Best Time to Buy and Sell Stock with Cooldown
这道题就相对要复杂了,因为涉及到cooldown一天,所以实际上要把一个状态拆成两个,然后额外加一个cooldown状态,这样总共就是四个状态
对于buy来说可以之前是buy但是没buy的状态,之前可以是cooldown的状态,还可以之前是sell的状态.对于sell状态,可以之前是sell但是没sell的状态,可以是cooldown状态.对于sell today状态来说,之前必须是buy的情况.对于最后的cooldown状态,之前必须是sell todayz状态
本质上这道题要画出状态转换图即可
class Solution:
def maxProfit(self, prices: List[int]) -> int:
dp_hold=[0 for _ in range(len(prices))]
dp_sell=[0 for _ in range(len(prices))]
dp_sellT=[0 for _ in range(len(prices))]
dp_cool=[0 for _ in range(len(prices))]
dp_hold[0]=-prices[0]
for i in range(1,len(prices)):
dp_hold[i],dp_sell[i],dp_sellT[i],dp_cool[i]=max(dp_hold[i-1], dp_cool[i-1]-prices[i],dp_sell[i-1]-prices[i]),max(dp_sell[i-1],dp_cool[i-1]),dp_hold[i-1]+prices[i],dp_sellT[i-1]
return max(dp_cool[-1],dp_sell[-1],dp_sellT[-1])
总结: 本系列buy&sell主要的关键点在于画出状态转换图并且设定好状态即可,对于初值来说要考虑如何保持当前的0状态.dp本身的定义就是在当前状态下,最大的利益是多少