Python实现力扣刷题之打家劫舍III（337题），超简单，一看就会

题目描述

在上次打劫完一条街道之后和一圈房屋后，小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为“根”。 除了“根”之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫，房屋将自动报警。

计算在不触动警报的情况下，小偷一晚能够盗取的最高金额。

如下示例所示

原理阐述

1. 该题目考察的是二叉树。
2. 相邻节点不能选择的情况下，求所有节点相加的最大值。
   1. 换句话说，也就是根节点选择，则左右节点的值不选；根节点不选择，左右节点的值可以选择
3. 该问题使用分治思想最合适。在多层二叉树的情况下，将问题分解为某个节点是否要选择（偷），下面以节点K为例
   1. 选择（偷）：当偷节点K内金额时，K.left和K.right两个子节点金额不能偷。但是，K.left.left(right)和K.right.left(right)内的金额是可以偷的，也就是说隔层的金额是可以偷的
      1. 根据上面的思路，K.left.left.left(right)、K.left.right.left(right)和K.right.left.left(right)、K.right.right.left(right)又不可以选择
      2. 为了方便起见，我们使用g()函数来代表上述操作，即g(K.left)代表不选择K.left的情况下获得金额总和
      3. 我们又设定选择该节点的函数f()，即f(K)代表选择K节点情况下的金额总和
      4. 那么可以得到：$f(k)=g(k.left)+g(k.right)$
   2. 不选择（不偷）：不偷节点K内金额时，则可以选择偷或不偷K.left或者K.right内的金额。这个地方可能会有人有疑问，为什么不偷呢，肯定要偷啊，于是我举下面这个例子。
      1. 举例说明为什么可以不偷K.left(right)
         
         观察该二叉树，如果我们不选择根节点3，直接偷左节点数值或右节点数值，会发现我们得到的是4,5，而错失了下面的1000,1000,1000，这时候我们选择不偷更好。
      2. 于是不偷K节点时的最大金额$g(k)=\max (f(k.left),g(k.left))+\max (f(k.right),g(k.right))$

代码实现

#Definition for a binary tree node.
#class TreeNode:
#def __init__(self, val=0, left=None, right=None):
    #self.val = val
	#self.left = left
	#self.right = right
class Solution:
    def rob(self, root: TreeNode) -> int:
        #DP规划
        def _rob(root):
            if root is None:return 0,0

            ls, ln = _rob(root.left)
            rs, rn = _rob(root.right)

            return root.val + ln + rn , max(ls, ln) + max(rs, rn)
        return max(_rob(root))

举例实现代码操作

注意图中的带圆圈的数字代表实现顺序。