LeetCode每日一练——打家劫舍

示例：

在一个地区，只有一个入口，我们称之为“根”。除了“根”之外，每栋房子有且只有一个“父”房子与之相连。一番侦查后，小偷意识到，这里的房屋排列类似于一棵二叉树。如果两个直接相连的房子在同一天晚上被打劫，房屋将自动报警。计算在不触动警报的情况下，小偷一晚能够盗取的最高金额。

输入：[3,2,3,null,3,null,1]
  3
 / \
2   3
 \    \
  3    1
输出：7
解释：3+3+1


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

  3
 / \
4   5 
/ \   \ 
1   3   1
输出：9
解释：4+5

深度遍历

# -*- coding:UTF-8 -*-
class TreeNode():
	def __init__(self,x,left=None,right=None):
		self.val = x
		self.left = left
		self.right = right
	
class Solution:
    def rob(self,root):
    	def dfs(root):
    		if root==None:
    			return 0,0
	    	left = dfs(root.left)
	    	right = dfs(root.right)
	    	childAmt = max(left)+max(right)#不要当前节点
	    	msAmt = root.val + left[0]+right[0]#要当前节点
	    	return childAmt,msAmt
	    return max(dfs(root))  
    

treeNode=TreeNode(3,TreeNode(2,None,TreeNode(3)),TreeNode(3,None,TreeNode(1)))    	
result = Solution()
print(result.rob(treeNode))