LeetCode 213.打家劫舍 II-CSDN博客

本文链接：https://blog.csdn.net/zxl1990_ok/article/details/127240417

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

解题思路

思路一: 动态规划

与LeetCode 198：打家劫舍题目区别在于房屋是首尾相连的。所以在房屋数大于等于2时，第一间房屋和最后一间房屋不能在同一晚上偷窃。

如何解决这个问题呢？
假设房屋数为k, 如果不偷最后一间房屋，则偷窃房屋的范围为[0, k - 2]; 如果不偷第一间房屋，则偷窃房屋的范围为[1, k - 1]。我们可以对这两段范围分别计算可以偷的最高总金额，其中的最大值即为答案。

用 dp[i] 表示前 i 间房屋能偷窃到的最高总金额，那么就有如下的状态转移方程：
dp[i]=max(dp[i−2] + nums[i], dp[i−1])

边界条件为：

dp[0]=nums[0]               // 只有一间房屋，则偷窃该房屋  
dp[1]=max(nums[0],nums[1])  // 只有两间房屋，选择其中金额较高的房屋进行偷窃

返回值：
返回 dp 列表最后一个元素值，即所有房间的最大偷窃价值。

简化空间复杂度：
我们发现 dp[k] 只与 dp[k - 1] 和 dp[k - 2] 有关系，因此我们可以设两个变量 curr和 prev 交替记录，将空间复杂度降到 O(1)

实现代码如下：

/**
 * @param {number[]} nums
 * @return {number}
 */
var rob = function(nums) {
    let length = nums.length;
    if (length == 1) {
        return nums[0];
    } else if (length == 2) {
        return Math.max(nums[0], nums[1]);
    }
    const robRange = function (start, end) { 
        // 循环开始时，curr 表示 dp[k - 1]，prev 表示 dp[k - 2]
        let prev = nums[start], curr = Math.max(nums[start], nums[start + 1]);
        // 每次循环，计算“偷到当前房子为止的最大金额”
        for (let i = start + 2; i <= end; i++) {
            let temp = Math.max(curr, prev + nums[i]);
            prev = curr;
            curr = temp;
        }
        // 循环结束时，curr 表示 dp[k]，prev 表示 dp[k - 1]
        return curr;
    }
    return Math.max(robRange(0, length - 2), robRange(1, length - 1));
};