Leetcode Day20 打家劫舍

198 最基础

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 二叉树中的打家劫舍

class Solution:def rob(self, root: Optional[TreeNode]) -> int:def dfs(node: Optional[TreeNode]) -> (int, int):if node is None:  # 递归边界return 0, 0  # 没有节点，怎么选都是 0l_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_robreturn max(dfs(root))  # 根节点选或不选的最大值

注意下not_rob是 max(左边偷不偷) + max(右边偷不偷)