代码随想录 | 动态规划 part9

198 打家劫舍

这问题并不是只有隔一个抢一个两种方式,因为可以隔很多个抢一个实现最大收益
背包问题,容量最大是所有家庭的金钱总和(不可能超过,最多相等)
物品是家庭,但是禁止连续拿相邻的两个物品
dp表示能抢到的最大金额
每家只能抢一次,01背包,无所谓顺序,先遍历物品再遍历背包,且内层循环从大到小
写不出来,陷入背包思维定式了
动态规划的思想就是根据前一个状态递归当前状态
考虑影响当前dp的关键因素:是否偷当前的房子
只跟前一个和前两个dp有关

class Solution:
    def rob(self, nums: List[int]) -> int:
        if len(nums) == 1: return nums[0]
        dp = [0]*len(nums)
        dp[0] = nums[0]
        dp[1] = max(nums[:2])
        for i in range(2,len(nums)):
            dp[i] = max(dp[i-2]+nums[i],dp[i-1])
        return dp[-1]

213 打家劫舍2

与原题只有一处变化,不允许同时抢劫第一个和最后一个房子
改变初始化策略,从最大的一个开始遍历,如此必然不会偷最后一个
提交发现思路有误,不能包含所有的情况
看了解答,居然是最简单粗暴的跑两遍比大小

class Solution:
    def rob(self, nums: List[int]) -> int:
        if len(nums)==1: return nums[0]
        res1 = self.dp(nums[:-1])
        res2 = self.dp(nums[1:])
        # print(res1,res2)
        return max(res1,res2)

    def dp(self,nums):
        if len(nums) == 1: return nums[0]
        dp = [0]*len(nums)
        dp[0] = nums[0]
        dp[1] = max(nums[:2])
        for i in range(2,len(nums)):
            dp[i] = max(dp[i-1],dp[i-2]+nums[i])
        return dp[-1]

还是得注意输入的特殊情况!!

337 打家劫舍3

树形dp,结合了二叉树的遍历和dp递推公式
先确定一下用什么遍历方式?
根节点 = max(左孩子+右孩子,自身)
后序遍历?
没写出来,看了解答发现每个dp元素需要存储两个数值,一个偷当前节点一个不偷当前节点的最大金额

class Solution:
    def rob(self, root: Optional[TreeNode]) -> int:
        return max(self.recur(root))

    def recur(self, root):
        if root == None:
            return [0,0]
        left = self.recur(root.left)
        right = self.recur(root.right)
        return [left[1]+right[1]+root.val,max(left)+max(right)]

还有一个点是,当不偷根节点时,左右孩子可偷,但是具体偷不偷得取最大的,不能无脑选左右孩子都偷

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值