LeetCode 337: 打家劫舍III

题目描述

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

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

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

题目分析

用dfs算出只从根节点开始抢的，和不从根节点开始抢的，比较大小即可。

Max(dfs(root), dfs(root.left) + dfs(root.right))

注意DFS实质为递归，时间复杂度为递归的时间复杂度和递归的次数，上面的描述中递归次数太多了，容易导致超时，我们便可以把尝试过的节点缓存起来。因为如果抢根节点，则需要把孙子节点的最大返回值计算出来，当孙子节点为根节点时，这个值又需要再被计算一次，所以我们便可以把每个节点为根节点时，能抢到的最大钱存起来，就不需要再重复递归下去。建立一个HashMap<TreeNode, Integer> memo, 键值对entity里面存放的是map[当前节点：当前节点为根节点能抢到的最大钱数]。

代码实现

class Solution {
    public int rob(TreeNode root) {
       return dfs(root, new HashMap<>());
    }

    public int dfs(TreeNode root, HashMap<TreeNode, Integer> memo) {
        // 1 截止条件
        // 1.1 为叶子节点，返回0
        if(root == null) {
            return 0;
        }
        // 1.2 为缓存节点，返回以该节点为根节点能偷到的钱
        if(memo.containsKey(root)) {
            return memo.get(root);
        }
		
		// 选择偷根节点
        int money = root.val;

        // 2 选择条件
        // 2.1 左子树不为空
        if(root.left != null) {
            money += dfs(root.left.left, memo) + dfs(root.left.right, memo);
        }
        // 2.2 右子树不为空
        if(root.right != null) {
            money += dfs(root.right.right, memo) + dfs(root.right.left, memo);
        }
        money = Math.max(money, dfs(root.right, memo) + dfs(root.left, memo));
        // 注意这里要存进去比较后的更大money
        memo.put(root, money);
        return money;
    }

}