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  # 没有节点，怎么选都是 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(右边偷不偷)