leetcode 337. 打家劫舍 III

最新推荐文章于 2023-05-25 11:43:47 发布

万古星空悬皓月

最新推荐文章于 2023-05-25 11:43:47 发布

阅读量180

点赞数

分类专栏：刷题 leetcode题解

本文链接：https://blog.csdn.net/qq_38043440/article/details/88412626

版权

刷题同时被 2 个专栏收录

101 篇文章 0 订阅

订阅专栏

leetcode题解

43 篇文章 0 订阅

订阅专栏

本文针对LeetCode上的打家劫舍III问题，提供了一种基于动态规划的解决方案。通过定义状态f(root,0)为在可选root节点情况下盗取的最高金额，f(root,1)为不可选root节点情况下的最高金额，文章详细阐述了状态转移方程，并附带Python代码实现。

摘要由CSDN通过智能技术生成

题目链接 https://leetcode-cn.com/problems/house-robber-iii/comments/
题目描述
1. 在上次打劫完一条街道之后和一圈房屋后，小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为“根”。除了“根”之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。如果两个直接相连的房子在同一天晚上被打劫，房屋将自动报警。
  
  计算在不触动警报的情况下，小偷一晚能够盗取的最高金额。
2. 示例 1:
```
输入: [3,2,3,null,3,null,1]

     3
    / \
   2   3
    \   \ 
     3   1

输出: 7 
解释: 小偷一晚能够盗取的最高金额 = 3 + 3 + 1 = 7.
```
  示例 2:
```
输入: [3,4,5,1,3,null,1]

     3
    / \
   4   5
  / \   \ 
 1   3   1

输出: 9
解释: 小偷一晚能够盗取的最高金额 = 4 + 5 = 9.
```
解题思路
1. 动态规划
2. 状态定义：f(root, 0) 表示 root在能选择该节点的情况下盗取的最高金额，：f(root, 1) 表示 root在不能选择该节点的情况下盗取的最高金额；
3. 状态转移方程 $f(root, 1) = f(root.left, 0) + f(root.right, 0)$

代码

python

class Solution:
    def rob(self, root: TreeNode) -> int:
        vis = {} 
        def _dfs(root, can_chice):
            if not root: return 0
            nonlocal vis
            if (root, can_chice) in vis:return vis[(root, can_chice)]
            chice = can_chice == 0 and (root.val + _dfs(root.left, 1) + _dfs(root.right, 1))
            no_chice = _dfs(root.left, 0) + _dfs(root.right, 0)
            vis[(root, can_chice)] = max(chice, no_chice)
            return vis[(root, can_chice)]
        return _dfs(root, 0)