打家劫舍问题

198. 打家劫舍

分析:
dp[i]:前i个房间(包括房间i)可以盗窃到的最大金额
对第i个房间,可以选择不偷,也可以选择偷 两种情况取最大
①偷:获得的金额等于当前房子的金额+前i-2个房间可以得到的最大金额(不是i-1是因为相邻的不能偷)
②不偷:获得的金额等于前i-1个房子能获得的金额

递推公式: dp[i] = max(dp[i-2]+nums[i], dp[i-1])
初始化:dp[0], dp[1]
遍历顺序:从小到大

class Solution:
    def rob(self, nums: List[int]) -> int:
        if len(nums)==1:
            return nums[0]
        if len(nums)==2:
            return max(nums[0],nums[1])

        dp = [0]*len(nums)
        dp[0] = nums[0]
        dp[1] = max(nums[0],nums[1])

        # 递推:
        for i in range(2,len(nums)):
            dp[i] = max(dp[i-2]+nums[i], dp[i-1])

        return dp[-1]

213. 打家劫舍 II

在Ⅰ的基础上改变:房间是环形的,首尾不能都偷
解决办法:两种方式中选最大(不考虑首元素/不考虑尾元素——转换为线性打家劫舍)

class Solution:
    def rob(self, nums: List[int]) -> int:
        n = len(nums)
        if n==1:
            return nums[0]        
        return max(self.roblist(nums[:-1]), self.roblist(nums[1:]))

    # 线性的打家劫舍
    def roblist(self, nums: List[int]) -> int:
        n = len(nums)
        if n == 1:
            return nums[0]
        if n == 2:
            return max(nums[0], nums[1])

        dp = [0]*n
        dp[0] = nums[0]
        dp[1] = max(nums[0], nums[1])
        for i in range(2,n):
            dp[i] = max(dp[i-1], dp[i-2]+nums[i])
        print(dp)
        return dp[-1]

337. 打家劫舍 III

树形
需要使用递归对节点遍历——后序(当前的节点的值取决于其孩子的值)
每个节点的返回值应该是两个,偷和不偷获得的最大利益

中间节点
偷:不偷左子数根节点获得的最大金额+不偷右子数根节点获得的最大金额+根节点的值
不偷:偷/不偷左子数根节点获得的最大金额+偷/不偷右子数根节点获得的最大金额 (左右可以偷,具体偷不偷要选最大的

class Solution:
    def rob(self, root: Optional[TreeNode]) -> int:  
        result = self.Traversal(root)
        return max(result[0], result[1])

    def Traversal(self, root):
        if root == None:
            return [0,0]  # 返回的是偷与不偷的量
        
        left = self.Traversal(root.left)  # 左
        right = self.Traversal(root.right)  # 右
        
        # 偷当前节点
        val1 = left[1] +right[1] + root.val

        # 不偷当前节点
        val2 = max(left[0],left[1]) +max(right[0],right[1])

        return [val1,val2]
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值