打家劫舍 III Python解法

小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为 root 。

除了 root 之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果 两个直接相连的房子在同一天晚上被打劫 ，房屋将自动报警。

给定二叉树的 root 。返回 在不触动警报的情况下 ，小偷能够盗取的最高金额 。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/house-robber-iii
 

例：

 

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

解析：

只有一个父节点，所以是每一层进行比较。使用后序遍历，从底部开始，因为相邻层不能同时“偷”，所以相邻层进行比较，从下往上，将最大值向上传递，类似DP数组。子树的最大值附加到当前节点的值上进行后序判断。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def rob(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        if root == None:  # 判空向上返回
            return 0
        self.rob(root.left)  # 后序遍历先遍历左子树
        self.rob(root.right)  # 后序遍历第二遍历右子树
        val1, val2 = (0 if root.left == None else root.left.val)+(0 if root.right == None else root.right.val), root.val  # val1是下一层的值，val2是当前判断节点的值
        if root.left != None:  # 当前节点左节点存在
            if root.left.left != None:  # 孙子节点是否存在
                val2 += root.left.left.val  # 孙子节点加当前节点
            if root.left.right != None:
                val2 += root.left.right.val
        if root.right != None:
            if root.right.left != None:
                val2 += root.right.left.val
            if root.right.right != None:
                val2 += root.right.right.val
        root.val = max(val1, val2)  # 孙子节点加当前节点的值与字节点的值进行比较，将更大的值赋予当前节点的值，即将子树的最大值赋上来进行后序判断
        return root.val  # 返回更改后的当前节点的值