leetcode 213. 打家劫舍 II

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

计算在不触动警报的情况下,小偷一晚能够盗取的最高金额。

示例 1:

输入: [3,2,3,null,3,null,1]

3
/ \
2 3
\ \
3 1

输出: 7
解释: 小偷一晚能够盗取的最高金额 = 3 + 3 + 1 = 7.
示例 2:

输入: [3,4,5,1,3,null,1]

  3
 / \
4 5
/ \ \
1 3 1

输出: 9
解释: 小偷一晚能够盗取的最高金额 = 4 + 5 = 9.

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

 

后序遍历 + 动态规划
1:递归的方法返回一个数组,长度为2,分别记录其使用 当前节点获取的最大值array[0], 不使用当前节点获取的最大值array[1]。
2:获取左右节点的数组之后,分别计算array[0]和array[1]的和为a,b。
3:若是采用当前节点,则 array[0] = Math.max(b + node.val, a);
4:若不采用当前节点,则 array[1] = Math.max(a, b);
5:最后返回array中的最大值, return Math.max(arr[0], arr[1]);

/**
 * 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 {
    public int rob(TreeNode root) {
        int[] arr = find(root);
        return Math.max(arr[0], arr[1]);
    }

    private int[] find (TreeNode node) {
        if (node == null) {
            return new int[]{0, 0};
        }
        int[] arr1 = find(node.left);
        int[] arr2 = find(node.right);
        int a = arr1[0] + arr2[0];
        int b = arr1[1] + arr2[1];
        arr1[0] = Math.max(b + node.val, a);
        arr1[1] = Math.max(a, b);
        return arr1;
    }
}

posted @ 2021-07-06 17:47  旺仔古李  阅读(36)  评论(0编辑  收藏  举报