《python算法与数据结构2000讲》0337. 打家劫舍 III

《python算法与数据结构2000讲》0337. 打家劫舍 III

• 标签：树、深度优先搜索、动态规划、二叉树
• 难度：中等

题目大意

小偷发现了一个新的可行窃的地区，这个地区的形状是一棵二叉树。这个地区只有一个入口，称为「根」。除了「根」之外，每栋房子只有一个「父」房子与之相连。如果两个直接相连的房子在同一天被打劫，房屋将自动报警。

现在给定这个代表地区房间的二叉树，每个节点值代表该房间所拥有的金额。要求计算在不触动警报的情况下，小偷一晚上能盗取的最高金额。

解题思路

树形动态规划问题。

对于当前节点 cur，不能选择子节点，也不能选择父节点。所以对于一棵子树来说，有两种情况：

• 选择了根节点
• 没有选择根节点

1. 选择根节点

如果选择了根节点，则不能再选择左右儿子节点，这种情况下的最大值为：当前节点 + 左子树不选择根节点 + 右子树不选择根节点。

2. 不选择根节点

如果不选择根节点，则可以选择左右儿子节点，共四种可能：