第五十一天|dp-CSDN博客

本文链接：https://blog.csdn.net/weixin_43880126/article/details/130479638

今天的还是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本身的定义就是在当前状态下,最大的利益是多少