算法练习-打家劫舍 III

题目描述

题目描述：
在上次打劫完一条街道之后和一圈房屋后，小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为“根”。 除了“根”之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫，房屋将自动报警。

计算在不触动警报的情况下，小偷一晚能够盗取的最高金额。

示例 1:

输入: [3,2,3,null,3,null,1]

     3
    / \
   2   3
    \   \ 
     3   1

输出: 7 
解释: 小偷一晚能够盗取的最高金额 = 3 + 3 + 1 = 7.
示例 2:

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

     3
    / \
   4   5
  / \   \ 
 1   3   1

输出: 9
解释: 小偷一晚能够盗取的最高金额 = 4 + 5 = 9.

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/house-robber-iii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

• 思路1：最优子结构
  这种思路主要是使用递归暴力破解，这种思路的时间复杂度会比较大。
  我们在题目上可以知道，父子关系的节点不能同时偷，爷孙关系的节点就可以同时偷，同时二叉树只有左右两个孩子，一个爷爷最多 2 个儿子，4 个孙子（三层满二叉树的情况）。那么问题就可以转化为了比较偷爷爷+4个孙子的钱和2个儿子的钱谁更多的贪心问题。
  那么我们算法思路就可以这样

1. 算出爷爷和四个孙子的钱
   money1 =root.val+ rob(root.left.left) + rob(root.left.right)+rob(root.right.left) + rob(root.right.right)
2. 然后再去算2个儿子的钱
   money2=rob(root.left) + rob(root.right)
3. 最后就是取最多的那个
   return Math.max(money1, money2)
   具体代码如下面的思路一

• 思路2 记忆化 -解决重复子问题
  我们发现思路1的时间复杂度很大，因为这个算法中重复了很多工作，就好像在算偷爷孙节点的钱的时候，也会遇到儿子节点，思路1是选择跳过，然后等爷孙节点算完然后再重新遍历一遍，这太费事了。

  为什么遇到了儿子节点的时候，也一块把他也算了，然后记忆起来呢？？
  所以思路2就是把儿子节点放到缓存中，当我们需要用的时候就取出来再算，这样就可以减少了重新遍历树再算了。
  由于二叉树不适合拿数组当缓存，我们这次使用哈希表来存储结果，TreeNode 当做 key，能偷的钱当做 value，因此我们在思路1的代码中进行优化，得出思路2的代码，效率足足是思路1的200多倍。

代码

//c思路一：
public int rob(TreeNode root) {
    if (root == null) return 0;

    int money = root.val;
    if (root.left != null) {
        money += (rob(root.left.left) + rob(root.left.right));
    }

    if (root.right != null) {
        money += (rob(root.right.left) + rob(root.right.right));
    }

    return Math.max(money, rob(root.left) + rob(root.right));
}

//思路二
public int rob(TreeNode root) {
    HashMap<TreeNode, Integer> memo = new HashMap<>();
    return robInternal(root, memo);
}

public int robInternal(TreeNode root, HashMap<TreeNode, Integer> memo) {
    if (root == null) return 0;
    if (memo.containsKey(root)) return memo.get(root);
    int money = root.val;

    if (root.left != null) {
        money += (robInternal(root.left.left, memo) + robInternal(root.left.right, memo));
    }
    if (root.right != null) {
        money += (robInternal(root.right.left, memo) + robInternal(root.right.right, memo));
    }
    int result = Math.max(money, robInternal(root.left, memo) + robInternal(root.right, memo));
    memo.put(root, result);
    return result;
}