337. 打家劫舍 III

Difficulty: 中等

在上次打劫完一條街道之後和一圈房屋後，小偷又發現了一個新的可行竊的地區。這個地區只有一個入口，我們稱之為“根”。 除了“根”之外，每棟房子有且只有一個“父“房子與之相連。一番偵察之後，聰明的小偷意識到“這個地方的所有房屋的排列類似於一棵二叉樹”。 如果兩個直接相連的房子在同一天晚上被打劫，房屋將自動報警。

計算在不觸動警報的情況下，小偷一晚能夠盜取的最高金額。

示例 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.

Solution

Language: ****

# 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:
    d = {}
    return self.robHelper(root, d)
    
  def robHelper(self, root, d):
    if not root:
      return 0
    
    if root in d:
      return d.get(root)
    
    res = 0
    
    if root.left:
      res += self.robHelper(root.left.left, d) + self.robHelper(root.left.right, d)
    if root.right:
      res += self.robHelper(root.right.left, d) + self.robHelper(root.right.right, d)
      
    res = max(res + root.val, self.robHelper(root.left, d) + self.robHelper(root.right, d))
    d[root] = res
    
    return res