LeetCode 337. 打家劫舍 III

最新推荐文章于 2024-07-10 15:03:22 发布

Sasakihaise_

最新推荐文章于 2024-07-10 15:03:22 发布

阅读量386

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签：树上DP

本文链接：https://blog.youkuaiyun.com/Sasakihaise_/article/details/130559431

LeetCode 专栏收录该内容

440 篇文章

订阅专栏

该文章介绍了一个使用深度优先搜索(DFS)解决的树形动态规划问题——打家劫舍III。在遍历树节点时，计算选择和不选择当前节点时的最大价值，并用一个数组记录这两个值。最后返回根节点的两个最大价值之一作为结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

337. 打家劫舍 III

【树上DP】还是选或者不选的问题，但是这个需要在DFS的过程中去处理，所以需要用一个数组来返回当前节点选和没选的最大值。

class Solution {

    public int[] dfs(TreeNode node) {
        if (node == null) return new int[] {0, 0};
        int v = node.val;
        int[] left = dfs(node.left);
        int[] right = dfs(node.right);
        int not = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);
        return new int[] {not, v + left[0] + right[0]};
    }

    public int rob(TreeNode root) {
        int[] ans = dfs(root);
        return Math.max(ans[0], ans[1]);
    }
}