LeetCode每日一练——打家劫舍

示例：

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

输入：2,[[0,1]]
输出：true
解释：总共有2门课程。学习课程1之前，你需要完成课程0。所以这是可能的。

输入：2,[[1,0],[0,1]]
输出：false
解释：总共有2门课程。学习课程1之前，你需要完成课程0,并且学习课程0之前，你还要完成课程1，这是可能的。

深度遍历

# -*- 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))