题目
在上次打劫完一条街道之后和一圈房屋后,小偷又发现了一个新的可行窃的地区。这个地区只有一个入口,我们称之为“根”。 除了“根”之外,每栋房子有且只有一个“父“房子与之相连。一番侦察之后,聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫,房屋将自动报警。
计算在不触动警报的情况下,小偷一晚能够盗取的最高金额。
示例 1:
输入: [3,2,3,null,3,null,1]
输出: 7
解释: 小偷一晚能够盗取的最高金额 = 3 + 3 + 1 = 7.
示例 2:
输入: [3,4,5,1,3,null,1]
输出: 9
解释: 小偷一晚能够盗取的最高金额 = 4 + 5 = 9.
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/house-robber-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解决思路
为 root 每个结点都设置二维数组,值分别代表偷和不偷的情况值,细分为两种情况:
- 每个结点的偷值都是:左侧子结点的不偷值 + 右侧子结点的不偷值 + 该节点的值
- 每个结点的不偷值都是: 左侧子结点的最大值 + 右侧子结点的最大值
每种情况都考虑到,每个结点都影响最终结果。
- 时间复杂度O(n)
与树的深度有关。 - 空间复杂度O(n)
每个结点都有对应的二维数组,空间复杂度与递归深度有关,由树的深度影响。
提交代码
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def rob(self, root: TreeNode) -> int:
a = self.digui(root) # 设置root对应的二维数组[偷值, 不偷值]
return max(a[0], a[1]) # 返回两者最大值为最终结果
# 参数为root结点, digui方法root结点的二维数组[偷值, 不偷值]
def digui(self, root):
if not root:
return [0, 0]
left = self.digui(root.left) # root 左侧子结点的[偷值, 不偷值]
right = self.digui(root.right) # root右侧子结点的[偷值, 不偷值]
robValue = left[1] + right[1] + root.val # root 的选择偷的值
skipValue = max(left[0], left[1]) + max(right[0], right[1]) # root 的不偷值
return [robValue, skipValue]
学习总结
- 对树遍历结点可以开设数组空间来保存当递归至这一结果的两种考虑情况;
- 不同于之前数组时候的情况,树的结构普遍采取深度优先搜索的方法对每一结点进行遍历,更多是与其他数据结构相结合的方式求解。