337. House Robber III

一、题目

  1、审题 

  

  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;
	}

  

posted @ 2019-06-30 16:21  skillking2  阅读(44)  评论(0编辑  收藏