题目:
打家劫舍 II:你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。
思路:
因为首尾两家相连,因此在一些执行的过程中不应该同时包含首和尾,因此针对不包含首和尾的情况下,分别使用上一题的思路,然后看两种情况下的最大值即可。一开始是用不包含尾的例子来测,发现有的例子可以通过,有的通不过,并对结果进行了分析,明白了还存在不包含首的情况。
程序:
class Solution: def rob(self, nums: List[int]) -> int: if not nums: return 0 length = len(nums) if length == 1: return nums[0] if length == 2: return max(nums[0], nums[1]) if length == 3: return max(nums[0], nums[1], nums[2]) candidate1 = self.auxiliaryFunc(nums[0:length - 1]) candidate2 = self.auxiliaryFunc(nums[1:]) return max(candidate1, candidate2) def auxiliaryFunc(self, nums): length = len(nums) auxiliary = [-1] * (length + 1) auxiliary[0] = 0 auxiliary[1] = nums[0] for index in range(1, length): auxiliary[index + 1] = max(auxiliary[index], auxiliary[index - 1] + nums[index]) return auxiliary[-1]