337. 打家劫舍 III

 

分析：状态转移分为节点选择和不选择两种情况，分别记录某个节点选择和不选择的可以得到的最大值

　　选择节点node，则左右孩子节点都不可以选择 

　　不选择节点node，则左右孩子节点都可以选择，需要比较大小看是否选择

代码：

/**
 * 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 {
    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(node) = g(node.left) + g(node.right)
        f.put(node,node.val + g.getOrDefault(node.left,0) + g.getOrDefault(node.right,0));
        // 当前节点不选择 g(node) = max(f(node.left),g(node.left)) + max(f(node.right),g(node.right));
        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)));
    }

}

 

 关键使用hash表作为散列函数存储状态值