leecode337. 打家劫舍 III

最新推荐文章于 2022-09-02 13:09:24 发布

原创最新推荐文章于 2022-09-02 13:09:24 发布 · 271 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#大数据

给定一个二叉树的根节点，小偷试图在不触发警报的情况下窃取最大的金额。每晚小偷可以进入一个节点，但相邻节点不能在同晚被窃取。该问题使用了记忆化搜索策略，通过递归地访问每个节点并记录选择或不选择节点的最大价值来解决。最终返回的是最大盗窃金额。

小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为 root 。

除了 root 之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。如果两个直接相连的房子在同一天晚上被打劫，房屋将自动报警。

给定二叉树的 root 。返回在不触动警报的情况下，小偷能够盗取的最高金额。

示例 1:

输入: root = [3,2,3,null,3,null,1]
输出: 7
解释: 小偷一晚能够盗取的最高金额 3 + 3 + 1 = 7
示例 2:

输入: root = [3,4,5,1,3,null,1]
输出: 9
解释: 小偷一晚能够盗取的最高金额 4 + 5 = 9

提示：

树的节点数在 [1, 104] 范围内
0 <= Node.val <= 104

利用记忆化搜索

class Solution {
    public int rob(TreeNode root) {
        Map<Cache, Integer> map = new HashMap<>();
        return Math.max(root.val + dfs(root.left, true, map) + dfs(root.right, true, map), dfs(root.left, false, map) + dfs(root.right, false, map));
    }

    public int dfs(TreeNode node, boolean parentSelect, Map<Cache, Integer> map) {
        if (node == null) {
            return 0;
        }
        Cache cache = new Cache(node, parentSelect);
        if (parentSelect) {//父节点选了，下面两个都不能选了
            if (!map.containsKey(cache)) {
                map.put(cache, dfs(node.left, false, map) + dfs(node.right, false, map));
            }

        } else {
            if (!map.containsKey(cache)) {
                //选此节点
                int cnt = node.val + dfs(node.left, true, map) + dfs(node.right, true, map);
                //不选此节点
                int cnt1 = dfs(node.left, false, map) + dfs(node.right, false, map);
                map.put(cache, Math.max(cnt1, cnt));
            }
        }
        return map.get(cache);
    }


    class Cache {
        TreeNode node;
        boolean parentSelect;

        public Cache(TreeNode node, boolean parentSelect) {
            this.node = node;
            this.parentSelect = parentSelect;
        }

        public TreeNode getNode() {
            return node;
        }

        public void setNode(TreeNode node) {
            this.node = node;
        }

        public boolean isParentSelect() {
            return parentSelect;
        }

        public void setParentSelect(boolean parentSelect) {
            this.parentSelect = parentSelect;
        }

        @Override
        public boolean equals(Object o) {
            if (this == o) return true;
            if (!(o instanceof Cache)) return false;
            Cache cache = (Cache) o;
            return parentSelect == cache.parentSelect && Objects.equals(node, cache.node);
        }

        @Override
        public int hashCode() {
            return Objects.hash(node, parentSelect);
        }
    }
}