【Leetcode】213- 打家劫舍II

问题简介

你是一个专业的小偷，计划偷窃沿街的房屋，每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈，这意味着第一个房屋和最后一个房屋是紧挨着的。同时，相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警 。
给定一个代表每个房屋存放金额的非负整数数组，计算你 在不触动警报装置的情况下 ，今晚能够偷窃到的最高金额。

示例 1：
输入：nums = [2,3,2]
输出：3
解释：你不能先偷窃 1 号房屋（金额 = 2），然后偷窃 3 号房屋（金额 = 2）, 因为他们是相邻的。

示例 2：
输入：nums = [1,2,3,1]
输出：4
解释：你可以先偷窃 1 号房屋（金额 = 1），然后偷窃 3 号房屋（金额 = 3）。
偷窃到的最高金额 = 1 + 3 = 4 。

示例 3：
输入：nums = [1,2,3]
输出：3

提示：
1 <= nums.length <= 100
0 <= nums[i] <= 1000

思路分析

本问题是在问题【Leetcode】198- 打家劫舍基础上新增了一个限制——第一个房屋和最后一个房屋是紧挨着的。这意味着第一个房屋和最后一个房屋是只能二选一的，意味着偷窃范围只能为：第一个房屋到倒数第二个房屋，或者第二个房屋到最后一个房屋。那么分别对两个范围的房屋计算最大价值（复用打家劫舍：使用动态规划算法计算一个序列中不相邻的元素之和最大值的算法），然后比较两个结果即可得最终最大价值。

代码实现

class Solution:
    def rob(self, nums: List[int]) -> int:
        if len(nums) == 1: return nums[0]
        if len(nums) == 2: return max(nums) 
        # ppre_0, pre_0 指代范围[0, len(nums]-2]中的“前两个”房屋；ppre_1, pre_1 指代范围[1, len(nums)-1]中的“前两个”房屋
        ppre_0, pre_0, ppre_1, pre_1 = nums[0], max(nums[0:2]), nums[1], max(nums[1:3])
        for i in range(2, len(nums)-1):
            ppre_0, pre_0 = pre_0, max(ppre_0 + nums[i], pre_0)
        for j in range(3, len(nums)):
            ppre_1, pre_1 = pre_1, max(ppre_1 + nums[j], pre_1)
        return max(pre_0, pre_1)