LeetCode.337. 打家劫舍 III

最新推荐文章于 2022-09-10 00:25:15 发布

什么是快了星球

最新推荐文章于 2022-09-10 00:25:15 发布

阅读量99

点赞数

分类专栏： LeetCode 文章标签： leetcode 深度优先算法

本文链接：https://blog.csdn.net/qq_39076073/article/details/124651696

版权

LeetCode 专栏收录该内容

17 篇文章 0 订阅

订阅专栏

小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为 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

public class daijashe3 {
    Map<TreeNode,Integer> memo = new HashMap<>();

    public int rob(TreeNode root){
        if(root==null) return 0;
        //利用备忘录消除重叠子问题
        if(memo.containsKey(root)) return memo.get(root);
        //抢，去下下家
        int do_it = root.val + (root.left == null? 0:rob(root.left.left)+rob(root.left.right))
                                +(root.right == null?0:rob(root.right.left)+rob(root.right.right));
        //不抢，去下家
        int not_do_it = rob(root.left) +rob(root.right);
        int res = Math.max(do_it,not_do_it);
        memo.put(root,res);
        return res;
    }
}