leetcode 337:打家劫舍III
小偷又发现了一个新的可行窃的地区。这个地区只有一个入口,我们称之为
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
Related Topics
树
深度优先搜索
动态规划
二叉树
思路1:深度优先遍历
分析:每个节点的最大金额有2种情况
- 左右孩子节点金额的最大值相加。
- 本节点的金额加上孩子的孩子的最大金额。
那么意味着要计算一个节点的最大金额,必须知道孩子节点的最大金额,也必须知道孩子的孩子节点的最大金额。可以使用数组返回这个结果。
递归函数dfs:int[] (TreeNode root)
nums[0]表示当前节点的最大值,nums[1]表示孩子节点的金额总和
- 如果root为null,直接返回
- 如果root的左右孩子为空,返回root.val和0
- 递归左右孩子计算左右孩子的金额。
- left[]
- right[]
- 那么当前节点的最大金额就是
- 如果选择当前节点,金额就是当前节点的值加上左右孩子的孩子(也就是在当前节点的下面第二层)的最大金额,就是
root.v
- 如果选择当前节点,金额就是当前节点的值加上左右孩子的孩子(也就是在当前节点的下面第二层)的最大金额,就是