leetcode-动态规划【打家劫舍】

最新推荐文章于 2024-08-15 20:27:56 发布

SunYutong_1234

最新推荐文章于 2024-08-15 20:27:56 发布

阅读量74

点赞数

分类专栏： leetcode 文章标签： python leetcode 数据结构动态规划算法

本文链接：https://blog.csdn.net/SunYutong_1234/article/details/124382598

版权

leetcode 专栏收录该内容

15 篇文章 2 订阅

订阅专栏

本文通过三道不同难度的打家劫舍问题，展示了动态规划在解决连续序列中最优选择问题上的应用。在第一题中，基本的动态规划思路是建立dp数组，从前往后更新最大收益。第二题引入了选择子区间的问题，需要考虑是否抢劫首尾房屋。第三题将问题拓展到树形结构，通过递归方式实现树形动态规划，计算节点的最大收益。这些题目揭示了动态规划在解决复杂优化问题时的灵活性和有效性。

摘要由CSDN通过智能技术生成

198. 打家劫舍

class Solution(object):
    def rob(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if len(nums) == 1:
            return nums[0]
        elif len(nums) == 2:
            return max(nums[0], nums[1])
        # dp数组记录当考虑第i个房屋的时候最大rob的值
        dp = [0 for _ in range(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

class Solution(object):
    def rob(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if len(nums) == 1:
            return nums[0]

        # 返回在nums[start:end+1]中rob到的最大的值
        def robRange(start, end):
            if start == end:
                return nums[start]
            dp = [0 for _ in range(len(nums))]
            dp[start] = nums[start]
            dp[start + 1] = max(nums[start], nums[start + 1])
            for i in range(start + 2, end + 1):
                dp[i] = max(dp[i - 1], dp[i - 2] + nums[i])
            return dp[end]

        res1 = robRange(0, len(nums) - 2)  # 不rob最后一个房子的情况
        res2 = robRange(1, len(nums) - 1)  # 不rob第一个房子的情况
        return max(res1, res2)

337. 打家劫舍iii

# Definition for a binary tree node.
class TreeNode(object):
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right


class Solution(object):
    def rob(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """

        def robTree(curNode):
            if not curNode:
                return [0, 0]
            left = robTree(curNode.left)
            right = robTree(curNode.right)
            # 不rob当前节点，rob或不rob左节点，rob或不rob右节点
            res0 = max(left) + max(right)
            # rob当前节点，则不能rob左右节点
            res1 = left[0] + right[0] + curNode.val
            return [res0, res1]

        res = robTree(root)
        return max(res[0], res[1])