LeetCode - 213. House Robber II

最新推荐文章于 2024-09-19 06:01:52 发布

shenzhu0127

最新推荐文章于 2024-09-19 06:01:52 发布

阅读量357

点赞数

分类专栏： leetcode 文章标签： leetcode Dynamic Programming

本文链接：https://blog.csdn.net/shenzhu0127/article/details/51870112

版权

leetcode 专栏收录该内容

138 篇文章 0 订阅

订阅专栏

这道题目是House Robber的升级版，在这一题中，房子首尾相连，所以头部的房子和尾部的房子不能同时抢劫，这道题目只是比上道House Robber题目多了这一个限制条件，在上道题目中，第一个房子可以随意选择抢劫或者不抢劫，而这道题目里第一个房子是否抢劫受到最后一个房子是否被抢劫的限制。由于题目的大致结构还是一样的，所以我们洗完更能够利用上道题目的解法来处理这个题目，经过思考后我们可以发现，这道题目可以分为两个House Robber题目，既然头部的房子和尾部的房子不能同时被抢劫，那么我们将题目分为两种情况，即不包括第一个房子，可以包括最后一个房子，和可以包括第一个房子，但是不包括最后一个房子。所以我们首先要做的就是将上道题目的解法更改，将其改为在一个范围内的rob，然后再调用两次这个函数取最大值即可。时间复杂度为O(n)，空间复杂度为O(1)，代码如下：

public class Solution {
    public int robRange(int[] nums, int start, int end){
        if(nums == null || nums.length == 0 || start > end) return 0;       // corner case
        
        int rob = 0;
        int notRob = 0;
        for(int i = start; i <= end; i++){
            int currRob = nums[i] + notRob;
            notRob = Math.max(rob, notRob);
            rob = currRob;
        }
        
        
        return Math.max(rob, notRob);
    }
    
    public int rob(int[] nums) {
        if(nums == null || nums.length == 0) return 0;
        if(nums.length == 1) return nums[0];
        if(nums.length == 2) return Math.max(nums[0], nums[1]);
        
        return Math.max(robRange(nums, 0, nums.length - 2), robRange(nums, 1, nums.length - 1));
    }
}

知识点：

1. 可以将环形的动态规划问题转化为两个线性的动态规划问题，要牢记这一思路，以后遇到相似的题目就按照这种方式处理