【Leetcode_总结】213. 打家劫舍 II - python

Q:

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

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


链接:https://leetcode-cn.com/problems/house-robber-ii/

思路:思路同打家劫舍,这道题与上一题的不同就是数组的开头和结尾不能同时出现,那么思路很简单了,分别计算包含开头不包含结尾和包含结尾不包含开头的两个数组的收益,取其最大值即可,核心代码不变

代码:

class Solution(object):
    def rob(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if not nums:
            return 0
        if len(nums) == 1:
            return nums[0]
        if len(nums) == 2:
            return max(nums)
        res1 = self.dp(nums[1:])
        res2 = self.dp(nums[:-1])
        return max(res1, res2)


    def dp(self,inputs):
        dp = [0 for _ in range(len(inputs)+1)]
        dp[1] = inputs[0]
        for i in range(2, len(inputs)+1):
            dp[i] = max(dp[i-1], dp[i-2]+inputs[i-1])
        return dp[-1]

【Leetcode_总结】213. 打家劫舍 II - python_第1张图片

你可能感兴趣的:(Leetcode)