LeetCode337打家劫舍 III(树形dp)

题目链接

第一次接触到树形dp的问题
本题对于某个结点进行分类讨论
1.当前结点拿，那么当前偷到的价值就是当前结点的价值加上左结点不拿时下面结点提供的价值和右节点不拿时下面结点提供的价值
2.当前结点不拿，那么当前偷到的价值就是左结点拿或者不拿的最大价值和右节点拿或者不拿的最大价值之和
总结一下
当前节点选择偷时，那么两个孩子节点就不能选择偷了
当前节点选择不偷时，两个孩子节点只需要拿最多的钱出来就行(两个孩子节点偷不偷没关系)
我们使用一个大小为 2 的数组来表示 int[] res = new int[2] 0 代表不偷，1 代表偷

   public int rob(TreeNode root) {
        int res[] = dp(root);
        return Math.max(res[0],res[1]);
    }
    public int[] dp(TreeNode root){
        if (root == null) return new int[2];
        // 分类讨论的标准是：当前结点偷或者不偷
        // 由于需要后序遍历，所以先计算左右子结点，然后计算当前结点的状态值
        int left[] = dp(root.left);
        int right[] = dp(root.right);
        int res[] = new int[2];
        // res[0]：以当前 node 为根结点的子树能够偷取的最大价值，规定 node 结点不偷
        // res[1]：以当前 node 为根结点的子树能够偷取的最大价值，规定 node 结点偷
        res[0] = Math.max(left[0],left[1])+Math.max(right[0],right[1]);
        res[1] = left[0]+right[0]+root.val;
        return res;
    }