337. House Robber III刷题笔记

最新推荐文章于 2024-11-01 15:19:21 发布

waiting不是违停

最新推荐文章于 2024-11-01 15:19:21 发布

阅读量150

点赞数

分类专栏： LeetCode Python刷题

本文链接：https://blog.csdn.net/weixin_44737922/article/details/125799350

版权

动态规划二叉树节点最大价值抢劫问题

关键词由CSDN通过智能技术生成

LeetCode Python刷题专栏收录该内容

221 篇文章 1 订阅

订阅专栏

树的dp问题

# 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: Optional[TreeNode]) -> int:
        res = self.rob_tree(root)
        return max(res)
    
    def rob_tree(self,node):
        if node is None:
            return (0,0)
        left = self.rob_tree(node.left)
        right = self.rob_tree(node.right)
        val1 = node.val + left[1] + right[1]
        val2 = max(left[0],left[1]) + max(right[0],right[1])
        return [val1,val2]