Leetcode刷题(第337题)——打家劫舍III

x33g5p2x  于2022-03-28 转载在 其他  
字(0.9k)|赞(0)|评价(0)|浏览(361)

一、题目

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

二、示例

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

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

三、思路
本题可分为两种情况,当前节点的值是偷还是不偷,所以我们用一个数组来表示,数组的第一位是偷,数组的第二位是不偷。并且本题我们采用从下向上进行访问。当前节点不存在时,则返回[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) {
    if(!root) return 0
    const rec = (node, isSeal, money) => {
        if(!node) return money
        let curMoney = money
        if(isSeal) {
            curMoney = curMoney + node.val
        }
        return rec(node.left, !isSeal, curMoney) + rec(node.right, !isSeal, curMoney)
    }
    return Math.max(rec(root, false, 0), rec(root, true, 0))
};

五、总结

相关文章