2021-08-06 ｜ 337. 打家劫舍 III 【二叉树跨层求路径最大和】

最新推荐文章于 2024-03-25 23:02:46 发布

奋进的杰西

最新推荐文章于 2024-03-25 23:02:46 发布

阅读量397

点赞数

分类专栏： Leetcode

本文链接：https://blog.youkuaiyun.com/weixin_41041275/article/details/119845336

版权

Leetcode 专栏收录该内容

304 篇文章

订阅专栏

在这里插入图片描述

# 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:
        def dfs(node):
            # 不选的结果，选的结果
            if not node:
                return [0,0]
            left = dfs(node.left)
            right = dfs(node.right)
            # 选择 就是左子树不选的结果+右子树不选的结果
            choose = node.val + left[0] + right[0]
            # 不选择 从子节点中选，比如左节点可以选择也可以不选择+右节点也可以选择不选择
            not_choose = max(left[0],left[1])+max(right[0],right[1])
            return [not_choose,choose]
        res = dfs(root)
        return max(res[0],res[1])