LeetCode-打家劫舍 III

算法记录

LeetCode 题目：

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

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

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

---

说明

一、题目

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

二、分析

• 和前两个打家劫舍一样货色，只是当前的偷盗变成了树形结构，也是自底向上的分析方式。
• 如果偷了当前的节点，那么左右子树的直接相连节点都不能偷。
• 如果没有偷当前节点，那么左右子树的直接相连节点可以不偷也可以偷，也就是取其中的最大值进行返回即可。
• 这就需要从底层返回两个值，一个是偷了子节点的最大值，一个是没有偷子节点的最大值，这样就可以进行动态转移了。

class Solution {
    public int rob(TreeNode root) {
        int[] ret = dfs(root);
        return Math.max(ret[0], ret[1]);
    }
    private int[] dfs(TreeNode root) {
        if(root == null) return new int[] {0, 0};
        int[] left = dfs(root.left);
        int[] right = dfs(root.right);
        int steal = root.val + left[1] + right[1];
        int unstwal = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);
        return new int[] {steal, unstwal};
    }
}
复制代码

---

总结

动态规划算法。