337. House Robber III

阳光穿透心脏的1/2处 2021-11-02 07:40 220阅读 0赞

一、题目

1、审题

![746986-20190630160050060-269074321.png][]

2、分析

给出一棵二叉树，取二叉树中的节点值，其中不能获取相邻层的二叉树节点。

二、解答

1、思路

方法一、

分两种情况， 情况一： root.val + root.left.left.val + root.left.right.val + root.right.left.val + root.right.right.val

情况二： root.left.val + root.right.val

然后取这两种情况的最大值，进行递归！

public int rob(TreeNode root) {
            if(root == null)
        		return 0;
        	
        	int val = 0;
        	if(root.left != null) 
        		val += rob(root.left.left) + rob(root.left.right);
        	
        	if(root.right != null)
        		val += rob(root.right.left) + rob(root.right.right);
        	
        	return Math.max(val + root.val, rob(root.left) + rob(root.right));
        }

方法二、

由于方法一中，每一次递归有很多重复的计算。采用 Map <Node， val> 记录当前节点对应的最大值！减少递归的重复计算！

public int rob(TreeNode root) {
            return robSub(root, new HashMap<TreeNode, Integer>());
    	}
    	
        private int robSub(TreeNode root, HashMap<TreeNode, Integer> map) {
        	if(root == null)
        		return 0;
        	if(map.containsKey(root))
        		return map.get(root);
        	
        	int val = 0;
        	if(root.left != null)
        		val += robSub(root.left.left, map) + robSub(root.left.right, map);
        	
        	if(root.right != null)
        		val += robSub(root.right.left, map) + robSub(root.right.right, map);
        	
        	val = Math.max(val + root.val, robSub(root.left, map) + robSub(root.right, map));
        	map.put(root, val);
        	return val;
    	}

转载于:https://www.cnblogs.com/skillking/p/11110254.html

[746986-20190630160050060-269074321.png]: /images/20211102/99371de7aa3a4d978165a7fb1ad9f2f0.png