LeetCode337. 打家劫舍 III(DFS记忆化搜索)

最新推荐文章于 2022-02-22 19:39:24 发布

Minkowski_lin

最新推荐文章于 2022-02-22 19:39:24 发布

阅读量280

点赞数 1

分类专栏： Python DFS LeetCode algorithm

本文链接：https://blog.csdn.net/qq_42991793/article/details/88780152

版权

Python 同时被 3 个专栏收录

22 篇文章 0 订阅

订阅专栏

LeetCode

17 篇文章 0 订阅

订阅专栏

algorithm

11 篇文章 0 订阅

订阅专栏

LeetCode337. 打家劫舍 III(DFS记忆化搜索)

# Definition for a binary tree node.
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution:
    def rob(self, root: 'TreeNode') -> 'int':
        ans = self.dfs(root)
        return max(ans[0], ans[1])

    def dfs(self, bt: TreeNode):
        ans = [0, 0]
        if not bt:
            return ans
        left = self.dfs(bt.left)
        right = self.dfs(bt.right)
        ans[0] = max(left[0], left[1]) + max(right[0], right[1])
        ans[1] = bt.val + left[0] + right[0]
        return ans