LeetCode刷题日记（337. 打家劫舍 III）

以当前节点为根，分别计算偷和不偷该节点时的最大金额，并递归考虑左右子树的情况.

'''
小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为 root 。
除了root之外，每栋房子有且只有一个“父“房子与之相连。
一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。
如果两个直接相连的房子在同一天晚上被打劫 ，房屋将自动报警。
给定二叉树的 root 。返回 在不触动警报的情况下 ，小偷能够盗取的最高金额 。
'''
'''
递归：
偷root节点，那么root.left和root.right就不能偷了
    money = root.val + money（不偷root.left + 不偷root.right）
不偷root节点，那么root.left和root.right就可以偷
    money = max(偷root.left , 不偷root.right）+max( 不偷root.left ,偷root.right)
'''

from typing import Optional, List
class Solution:
    def rob(self, root: Optional[TreeNode]) -> int:
        def rob0(root: Optional[TreeNode]) -> List[int]:
            # 返回一个列表，第一个元素表示偷root节点的最大金额，第二个元素表示不偷root节点的最大金额
            if not root:
                return 0, 0
            left = rob0(root.left)
            right = rob0(root.right)
            # 偷root节点，那么root.left和root.right就不能偷了
            money = root.val + left[1] + right[1]
            # 不偷root节点，那么root.left和root.right就可以偷
            not_money = max(left) + max(right)
            return money, not_money