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)

猜你喜欢

转载自blog.csdn.net/qfc_128220/article/details/131144477