LeetCode337 打家劫舍 III(中等 树上dp)

最佳买卖股票时机含冷冻期

1.题目

在上次打劫完一条街道之后和一圈房屋后，小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为“根”。 除了“根”之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫，房屋将自动报警。
计算在不触动警报的情况下，小偷一晚能够盗取的最高金额。

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/house-robber-iii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.题解

       本题其实是打家劫舍的扩展，将线性的动态规划换成了树上dp，需要求一颗树中不相邻的和的最大值，问题在于怎么解决这样的问题.
首先需要知道的是我们需要转化为这样一个问题dp[i][0] = Math.max(dp[i -1][0], dp[i - 1][1])，显然如果在一颗树上做想要这样计算是有难度的，我们首先考虑树有哪些遍历方式（前序、中序、后续）,即根左右，左根右，左右根。我们想要处理的是从子节点到父亲节点这样一个过程，那么很明显就可以使用后序遍历的方式来达到这样一个目的，也就是说，使用后序遍历的话，我们可以处理完子节点，然后再处理父节点，也是我们想要的递推公式。

回归到本题，当我们发现可以使用后序遍历的方式实现了递推的过程那么问题来了，具体的公式是怎么样的呢，也就是说当前这个节点有两种状态，选择和不选择，选择的话就是它的两个子节点没有选择的最大值加上val，而如果不选择的话就有两种情况了，分别是子节点选择的最大值和没有选择的最大值，分别求子节点的这两种情况的最大值即可。使用HashMap来保存这两种结果，最后返回根节点的结果既可以

• 代码如下

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    HashMap<TreeNode, Integer> choose = new HashMap<>();
    HashMap<TreeNode, Integer> nootChoose = new HashMap<>();
    public int rob(TreeNode root) {
        dfs(root);
        return Math.max(choose.getOrDefault(root, 0), nootChoose.getOrDefault(root, 0));
    }
    public void dfs(TreeNode node){
        if(node == null)
            return ;
        dfs(node.left);
        dfs(node.right);
        choose.put(node, nootChoose.getOrDefault(node.left, 0) + nootChoose.getOrDefault(node.right, 0) + node.val);
        nootChoose.put(node, Math.max(nootChoose.getOrDefault(node.left, 0), choose.getOrDefault(node.left, 0)) + Math.max(nootChoose.getOrDefault(node.right, 0), choose.getOrDefault(node.right, 0)));
    }
}