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

版权声明:本文为weixin_43579528原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。