LeetCode - 213 打家劫舍 II

目录

题目来源

题目描述

示例

提示

题目解析

算法源码

---

题目来源

213. 打家劫舍 II - 力扣（LeetCode）

题目描述

你是一个专业的小偷，计划偷窃沿街的房屋，每间房内都藏有一定的现金。这个地方所有的房屋都 围成一圈 ，这意味着第一个房屋和最后一个房屋是紧挨着的。同时，相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警 。

给定一个代表每个房屋存放金额的非负整数数组，计算你 在不触动警报装置的情况下 ，今晚能够偷窃到的最高金额。

示例

示例 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 打家劫舍_伏城之外的博客-CSDN博客拓展题

做本题之前，大家需要先把链接题目搞懂。

本题和链接题目的区别在于，198打家劫舍是线性数组，而213打家劫舍是环形数组。

线性数组，选择了nums[0]，对于nums[nums.length-1]的选择没有影响

环形数组，选择了nums[0]，则nums[nums.length-1]不能选，因为nums[0]和nums[nums.length-1]是相邻的。

因此，本题最简单的做法就是将环形数组首尾解开，变成线性数组。

即环形数组可以拆分为两类线性数组，

• 丢弃nums[0]，剩下部分组成线性数组
• 丢弃nums[nums.length - 1]，剩下部分组成线性数组

因此，本题只是相较于198多了一步环形数组拆成线性数组的动作，线性数组直接按照198的做法做即可。

JS算法源码

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

  return Math.max(
    stealMaxMoney(nums.slice(0, -1)),
    stealMaxMoney(nums.slice(1))
  );
};

function stealMaxMoney(nums) {
  const n = nums.length;

  const dp = new Array(n);

  if (n >= 1) dp[0] = nums[0];
  if (n >= 2) dp[1] = Math.max(nums[0], nums[1]);

  for (let i = 2; i < n; i++) {
    dp[i] = Math.max(dp[i - 1], dp[i - 2] + nums[i]);
  }

  return dp[n - 1];
}

打家劫舍 II - 提交记录 - 力扣（LeetCode）

 

Java算法源码

class Solution {
    public int rob(int[] nums) {
        if(nums.length == 1) return nums[0];

        return Math.max(stealMaxMoney(Arrays.copyOfRange(nums, 0, nums.length - 1)), stealMaxMoney(Arrays.copyOfRange(nums, 1, nums.length)));
    }

    public int stealMaxMoney(int[] nums) {
        int n = nums.length;

        int[] dp = new int[n];

        if (n >= 1) dp[0] = nums[0];
        if (n >= 2) dp[1] = Math.max(nums[0], nums[1]);

        for (int i = 2; i < n; i++) {
            dp[i] = Math.max(dp[i - 1], dp[i - 2] + nums[i]);
        }

        return dp[n - 1];
    }
}

打家劫舍 II - 提交记录 - 力扣（LeetCode）

Python算法源码

def stealMaxMoney(nums):
    n = len(nums)

    dp = [0] * n

    if n >= 1:
        dp[0] = nums[0]

    if n >= 2:
        dp[1] = max(nums[0], nums[1])

    for i in range(2, n):
        dp[i] = max(dp[i - 1], dp[i - 2] + nums[i])

    return dp[n - 1]


class Solution(object):
    def rob(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if len(nums) == 1:
            return nums[0]

        return max(stealMaxMoney(nums[:-1]), stealMaxMoney(nums[1:]))

打家劫舍 II - 提交记录 - 力扣（LeetCode）