【leetcode】337.打家劫舍III

题目

leetcode原题链接

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

示例 1:

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

示例 2:

输入: root = [3,4,5,1,3,null,1]
输出: 9
解释: 小偷一晚能够盗取的最高金额 4 + 5 = 9

提示：
树的节点数在 [1, 104] 范围内
0 <= Node.val <= 104

思路

• 房屋是树形结构，采用何种方式遍历遍历呢？由于要知道当前节点的最大金额，我们需要知道其左右子节点的最大金额，因此左右子节点要先遍历，中间节点后遍历，故采用后序遍历
• 后序遍历使用递归
  • 递归的返回值？返回该节点的偷与不偷两种金额结果，用长度为2的数组保存,[偷的金额，不偷的金额]
  • 递归结束条件？当遍历到空节点，返回[0,0]
  • 单层递归的逻辑？分两种情况
    • 偷当前节点获得的最大金额 = 左节点两个孩子节点的最大金额 + 右节点两个孩子节点的最大金额
    • 不偷当前节点的最大金额 = 左节点的最大金额（偷与不偷取最大的） + 右节点的最大金额（偷与不偷取最大的）

代码

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number}
 */
var rob = function(root) {

    function robTree(node){
        // 节点为空，偷与不偷都为0
        if(!node) return [0,0]
        let left = robTree(node.left)
        let right = robTree(node.right)
        // 偷当前节点
        let val1 = node.val + left[1] + right[1]
        // 不偷当前节点,左右节点可偷可不偷，结果最大就好
        let val2 = Math.max(left[0] , left[1]) + Math.max(right[0] , right[1])
        return [val1 , val2]
    }
    let res = robTree(root)
    return Math.max(res[0] , res[1])
};

复杂度

• 时间复杂度：$O(n)$
• 空间复杂度：$O(logn)$