代码随想录算法训练营第38天｜198.打家劫舍、213.打家劫舍II、337.打家劫舍 III

198.打家劫舍

leetcode 198.打家劫舍
代码随想录

class Solution:
    def rob(self, nums: List[int]) -> int:
        # dp[i] 表示，到第i家的时候，价值最大的
        # 选i和不选i
        # 选i就只能是dp[i-2] +  nums[i]
        # 不选i就是dp[i-1]
        # 也可以用二维数组，不过要注意，在不选当前屋子的时候，需要取前一个里面最大的。
        if len(nums) == 0:
            return 0
        if len(nums) == 1:
            return nums[0]
        
        dp = [0] * len(nums)
        dp[0] = nums[0]
        dp[1] = max(nums[0], nums[1])
        for i in range(2, len(nums)):
            dp[i] = max(dp[i-1], dp[i-2] + nums[i])
        return dp[-1]

213.打家劫舍II

leetcode 213.打家劫舍II
代码随想录

class Solution:
    def func(self, nums):
        if len(nums) == 1:
            return nums[0]
        dp = [0] * len(nums)
        dp[0] = nums[0]
        dp[1] = max(nums[0], nums[1])
        for i in range(2,len(nums)):
            dp[i] = max(dp[i-1], dp[i-2] + nums[i])
        return dp[-1]
    def rob(self, nums: List[int]) -> int:
        # 少考虑一个房间就能破解循环
        if len(nums) == 1:
            return nums[0]
        result1 = self.func(nums[:-1])
        result2 = self.func(nums[1:])
        return max(result1, result2)

337.打家劫舍 III

leetcode 337.打家劫舍 III
代码随想录

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def rob(self, root: Optional[TreeNode]) -> int:
        # 二维dp
        # 0表示拿，1表示不拿
        dp = self.traverse(root)
        return max(dp[0], dp[1])
    def traverse(self, root):
        if not root:
            return (0,0)
        left = self.traverse(root.left)
        right = self.traverse(root.right)
        # 当前节点拿，只能加左右孩子都不拿的情况
        val0 = root.val + left[1] + right[1]
        # 当前节点不拿,就是左右孩子最大的收益相加
        val1 = max(left[0], left[1]) + max(right[0], right[1])

        return (val0, val1)