DP树——打家劫舍III + 美团笔练

1、题目：

1）力扣打家劫舍III

2）美团笔试练题：

 

2、DP树

树DP，也就是在树上进行的DP。因为树固有的递归属性，所以树DP一般都是递归进行的。大致上，若要解一个给定问题，我们需要解其不同部分（即子问题），再根据子问题的解以得出原问题的解；而树的一些特点和性质有：

 

 

3、力扣打家劫舍分析

3.1题目描述

 3.2、算法分析

本题直观题意就是，一个二叉树，每个节点都有对应的权值，每个点都有选和不选两种状态，要求是直接相邻的两个节点不能同时被选（存在父子关系的不可以同时选），问能够选中点的最大权值是多少。

 

3.3 代码

class Solution {
    Map<TreeNode, Integer> f = new HashMap<TreeNode, Integer>();
    Map<TreeNode, Integer> g = new HashMap<TreeNode, Integer>();

    public int rob(TreeNode root) {
        dfs(root);
        return Math.max(f.getOrDefault(root, 0), g.getOrDefault(root, 0));
    }

    public void dfs(TreeNode node) {
        if (node == null) {
            return;
        }
        dfs(node.left);
        dfs(node.right);
        f.put(node, node.val + g.getOrDefault(node.left, 0) + g.getOrDefault(node.right, 0));
        g.put(node, Math.max(f.getOrDefault(node.left, 0), g.getOrDefault(node.left, 0)) + Math.max(f.getOrDefault(node.right, 0), g.getOrDefault(node.right, 0)));
    }
}

备注： map.getOrDefault(key ,defaultvalue)

        从map中取出key的值，如果不为null则正常取出，如果为null，则取defaultvalue的值

进阶优化：

 

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

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