198 最基础
python
class Solution:
def rob(self, nums: List[int]) -> int:
dp1 = [0] * len(nums)
dp2 = [0] * len(nums)
# dp1指第i天偷了, dp2指第i天没有偷
dp1[0] = nums[0]
for i in range(1, len(nums)):
dp1[i] = dp2[i - 1] + nums[i]
dp2[i] = max(dp1[i - 1], dp2[i - 1])
return max(dp1 + dp2)
213 房子是环形的, 最后一个和第一个相连
啊这, rob(nums[1:]) 和 rob(nums[:-1])中取大的就可以了
337 二叉树中的打家劫舍
python
class Solution:
def rob(self, root: Optional[TreeNode]) -> int:
def dfs(node: Optional[TreeNode]) -> (int, int):
if node is None: # 递归边界
return 0, 0 # 没有节点,怎么选都是 0
l_rob, l_not_rob = dfs(node.left) # 递归左子树
r_rob, r_not_rob = dfs(node.right) # 递归右子树
rob = l_not_rob + r_not_rob + node.val # 选
not_rob = max(l_rob, l_not_rob) + max(r_rob, r_not_rob) # 不选
return rob, not_rob
return max(dfs(root)) # 根节点选或不选的最大值
注意下not_rob是 max(左边偷不偷) + max(右边偷不偷)