32.动态规划-打家劫舍II-Leetcode 213(python)

  • 题目描述

你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警

给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。

  • 示例

示例 1:

输入: [2,3,2]
输出: 3
解释: 你不能先偷窃 1 号房屋(金额 = 2),然后偷窃 3 号房屋(金额 = 2), 因为他们是相邻的。

示例 2:

输入: [1,2,3,1]
输出: 4
解释: 你可以先偷窃 1 号房屋(金额 = 1),然后偷窃 3 号房屋(金额 = 3)。
     偷窃到的最高金额 = 1 + 3 = 4 。

  • 解决思路

这道题是在打家劫舍那道题的基础上添加了增加了限定条件,也就是把房屋连成一个圈,不能同时打劫第一个和最后一个,那么就可以分别求出,没有第一个房子时能偷到的金额最大值,和没有最后一个房子时能偷到的最大值。比较这两个中间较大的一个,即可。

打家劫舍:https://blog.csdn.net/Try_my_best51540/article/details/84502734

切片操作:

nums[1:] 切掉数组的第一个元素

nums[:-1]切掉数组的最后一个元素

  • 代码
class Solution(object):
    def rob(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        def sub_rob(nums):
            if len(nums) == 0:
                return 0
            elif len(nums) == 1:
                return nums[0]
            elif len(nums) == 2:
                return max(nums[0],nums[1])
          
            #固定长度的数组的定义
            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[len(nums)-1]
        
        n = len(nums)
        if n == 0: return 0
        if n <= 2: return max(nums)
        return max(sub_rob(nums[1:]),sub_rob(nums[:-1]))

猜你喜欢

转载自blog.csdn.net/Try_my_best51540/article/details/84815837