力扣—337打家劫舍III(记忆化递归)

题目描述

在上次打劫完一条街道之后和一圈房屋后，小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为“根”。 除了“根”之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫，房屋将自动报警。

计算在不触动警报的情况下，小偷一晚能够盗取的最高金额。

示例 1:

输入: [3,2,3,null,3,null,1]

     3
    / \
   2   3
    \     \ 
     3    1

输出: 7 
解释: 小偷一晚能够盗取的最高金额 = 3 + 3 + 1 = 7.

示例 2:

输入: [3,4,5,1,3,null,1]

     3
    / \
   4   5
  / \     \ 
 1   3   1

输出: 9
解释: 小偷一晚能够盗取的最高金额 = 4 + 5 = 9.

解题思路

记忆化递归+回溯

用root表示根节点，首先定义dfs(root)，表示从根节点开始遍历，用total来记录从根节点遍历的值，遍历一个节点后，需要隔一个才行。当不从根节点开始偷，从两边开始，因此还需要进行dfs(root.left) + dfs(root.right),

class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution(object):
    def __init__(self):
        self._dict = {}
    def rob(self, root):
        if not root:
            return 0
        def dfs(node):
            if not node:
                return 0
            if node in self._dict:
                return self._dict[node]
            total = node.val
            if node.left:
                total += (dfs(node.left.left) + dfs(node.left.right))
            if node.right:
                total += (dfs(node.right.left) + dfs(node.right.right))
            result= max(total, dfs(node.left) + dfs(node.right))
            self._dict[node] = result
            return result

        res = dfs(root)
        return res

S = Solution()
n1 = TreeNode(3)
n2 = TreeNode(4); n3 = TreeNode(5)
n4 = TreeNode(1); n5 = TreeNode(3); n6 = TreeNode(1)
n1.left = n2; n1.right = n3
n2.left = n4; n2.right = n5; n3.right = n6
print(S.rob(n1))