力扣-213.打家劫舍 II

本文介绍了一种基于动态规划的方法,解决打家劫舍问题,当房屋围成一圈时,特别指出13号房不可同时取。通过比较不同范围的劫掠结果,找出最终最优解。代码展示了如何利用状态转移方程求解并实现rob函数。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

在前一题力扣-198.打家劫舍_Node_Su的博客-优快云博客的基础上,设置房屋围成一圈,即三个的时候,1 3 不可同时取

围成一圈则考虑

打劫[0,n-1]的结果和[1,n]的结果哪个最优,即为最终最优解

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

    def rob_1(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        n = len(nums)

        dp = [0] * n  # 初始钱包为0
        dp[0] = nums[0]
        if n == 1:
            return nums[0]
        if n == 2:
            dp[1] = max(nums[0], nums[1])
            return dp[1]
        if n >= 3:
            dp[1] = max(nums[0], nums[1])
            dp[2] = max(nums[0] + nums[2], nums[1])
            for i in range(3, n):
                dp[i] = max(dp[i - 2] + nums[i], dp[i - 3] + nums[i])
            res = -1
            for i in range(n):
                res = max(dp[i], res)
            return res


if __name__ == '__main__':
    nums = [1, 2, 1, 1]
    Sol = Solution()
    res = Solution.rob(Sol, nums)
    print(res)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值