题目描述
在上次打劫完一条街道之后和一圈房屋后,小偷又发现了一个新的可行窃的地区。这个地区只有一个入口,我们称之为“根”。 除了“根”之外,每栋房子有且只有一个“父“房子与之相连。一番侦察之后,聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫,房屋将自动报警。
计算在不触动警报的情况下,小偷一晚能够盗取的最高金额。
示例 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))