LeetCode Everyday - 打家劫舍 II

开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第19天，点击查看活动详情

打家劫舍 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

解题思路:

动态规划，将上一题的数组变成了环形。

1. 将环拆成两个数组，因为不能同时抢首尾两个房子
2. 将两个数组分别抢一次，取其最大值
3. 考虑边界，只有一个房子的时候取该值

我的答案:

/**
 * @param {number[]} nums
 * @return {number}
 */
var rob = function(nums) {
  if (!nums.length) {
    return 0
  }
  if (nums.length === 1) {
    return nums[0]
  }

  var rob2 = function (nums) {
    if (!nums.length) {
      return 0
    }
    let dp = [] // 存放到第i个房间能得到的最高金额
    dp[0] = nums[0] // 到第0个房间最高金额nums[0]
    dp[1] = Math.max(nums[0], nums[1]) // 到第1个房间金额为max(nums[0],nums[1])
    for (let i = 2; i < nums.length; i++) {
      // 以后到最后一个房间有两个选择：保持上一个房间金额不变，或者取上上个金额+本房间，二者取最大值
      dp[i] = Math.max(dp[i - 1], dp[i - 2] + nums[i])
    }
    return dp[nums.length - 1] // 返回最后一个房间拿到的最高金额
  };

  // 将环拆成两个数组，因为不能同时抢首尾两个房子
  let arr1 = nums.slice(0, nums.length - 1)
  let arr2 = nums.slice(1, nums.length)
  // 将两个数组分别抢一次，取其最大值
  return Math.max(rob2(arr1), rob2(arr2))
};

最后

如果有更好的解法或者思路, 欢迎在评论区和我交流~ ღ( ´･ᴗ･` )