Java_多态动态规划_打家劫舍Ⅱ

80 阅读2分钟

题目链接: leetcode.cn/problems/Pz…

题目描述:

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

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

示例 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 = [0] 输出: 0

提示:

  • 1 <= nums.length <= 100
  • 0 <= nums[i] <= 1000

题目解析

本题与《按摩师》的算法原理相同,只不过需要注意房屋围成一圈,所以我们需要分类讨论了。 第一种是偷第一家,也就是 nums[0],那么第二家和最后一家都不能偷了。 第二种是不偷第一家,范围就变成了 nums[1] - nums[i - 1]。这样的话我们直接把核心代码写出来(核心代码与按摩师相同),在调用的时候直接传递参数范围就可以了。

代码

class Solution {
    public int rob(int[] nums) {
        int len = nums.length;
        int result = Math.max(nums[0] + rob1(nums,2,len - 2), rob1(nums,1,len - 1));
        return result;
    }
    public static int rob1(int[] nums, int begin, int end) {
        if (begin > end) return 0;
        int len = nums.length;
        int[] t = new int[len];
        int[] f = new int[len];
        t[begin] = nums[begin];
        for (int i = begin + 1; i <= end; i++) {
            t[i] = f[i - 1] + nums[i];
            f[i] = Math.max(f[i - 1], t[i - 1]);
        }
        return Math.max(t[end], f[end]);
    }
}