每日一道算法题(9)--打家劫舍III

最新推荐文章于 2024-11-16 21:36:47 发布

秃狼

最新推荐文章于 2024-11-16 21:36:47 发布

阅读量112

点赞数 1

分类专栏： Java

本文链接：https://blog.csdn.net/Ostkakah/article/details/119808768

版权

LeetCode 二叉树动态规划记忆化搜索打家劫舍

关键词由CSDN通过智能技术生成

Java 专栏收录该内容

55 篇文章 6 订阅

订阅专栏

题目

LeetCode :337

思路分析

思路基本上就与打家劫舍差不多，要注意的是因为是二叉树的形式，所以每次可以同时打劫两个房间(左和右),选择上还是分为打劫当前房间或不打劫当前房间，最终求出最大值。

代码展示

class Solution {
    Map<TreeNode, Integer> map = new HashMap<>(); 
    public int rob(TreeNode root) {
        if(root == null)
            return 0;
        if(map.containsKey(root)){//记忆化搜索
            return map.get(root);
        }
        //取
        int val1 = 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 val2 = rob(root.left) + rob(root.right);
        int result = (val1 >= val2) ? val1 : val2;
        map.put(root, result);
        return result;
    }
}