代码随想录二刷-打家劫舍II

213：打家劫舍II

链接: Leetcode213.打家劫舍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。

思路

1. 分析题干
   a. 首先明确，房屋是围成一圈的。所以，如果偷了最后一间房屋就不能偷第一间房屋。
   b. 我们的目标是求出最高金额。常规思路是在最后一间房屋的dp值收获结果(记为dp[end])，dp值由dp[end-1]和dp[end-2]+nums[i]决定。但问题来了，现在dp[end]还和dp[0]有关，这下麻烦了。为了解决这个问题，需要分类讨论，对nums数组做切割。随想录分成情况二和情况三。本文注重分析情况二和情况三背后的含义。
   c. 情况二是在直接放弃最后一间房屋的情况下，选择在倒数第二个房屋收获最终结果。既然不偷最后一间，那么第一间就在偷的考虑范围内，所以也可以把该情况归纳为考虑偷第一间的情况。
   d. 情况三是在直接放弃第一间房屋的情况下，选择在最后一间房屋收获最终结果。既然不偷第一间，那么最后一间就在偷的考虑范围内。
   e. 一个是放弃第一间，一个是考虑第一间，到底哪种方案更优，二者取最大。

2. 写题顺序
   a. 找出情况二的范围内偷取金额的最大值
   b. 找出情况三的范围内偷取金额的最大值
   c. 二者取最大

3. 注意事项
   a. 相比于“198.打家劫舍”，该题把第一间房屋也纳入了需要考虑前一间偷还是不偷的情况里，将问题难度升级

4. 代码

class Solution {
public:
    int rob(vector<int>& nums) {
        int n = nums.size();
        if(n <= 0) return 0;
        if(n == 1) return nums[0];
        
        int result1 = robRange(nums, 0, n-2);
        int result2 = robRange(nums, 1, n-1);
        return max(result1, result2);
    }
    int robRange(vector<int>& nums, int start, int end){
        if(end-start == 0) return nums[start];
        
        vector<int> dp(nums.size(), 0);
        dp[start] = nums[start];
        dp[start+1] = max(nums[start], nums[start+1]);
        for(int i = start+2; i <= end; ++i){
            dp[i] = max(dp[i-1], dp[i-2]+nums[i]);
        }
        
        return dp[end];
    }
};

参考: 代码随想录-213.打家劫舍II