【Leetcode】01背包问题-213. 打家劫舍 II

【Leetcode】01背包问题-213. 打家劫舍 II

题目

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

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

思路

参考这道题198.打家劫舍

https://leetcode.cn/problems/house-robber/

https://blog.csdn.net/qq_44653420/article/details/127111359?spm=1001.2014.3001.5501

这道题在原来的基础上添加了圆环限制，所以在原来的基础上，我们考虑两种情况：

• 情况一：从第一个元素到倒数第一个元素
• 情况二：从第二个元素到最后一个元素
• 将原本的198.打家劫舍代码中的0换成 start

代码

class Solution {
public:
    int rob(vector<int>& nums) {
        if(nums.size() == 0) return 0;

        if(nums.size() == 1) return nums[0];

        // 情况一：从第一个元素到倒数第一个元素
        int result1 = robRange(nums,0,nums.size() - 2);

        // 情况二：从第二个元素到最后一个元素
        int result2 = robRange(nums,1,nums.size() - 1);

        return max(result1,result2);
    }

    // 将原本的198.打家劫舍代码中的0换成 start
    int robRange(vector<int> &nums,int start,int end)
    {
        if(end == start) 
        {
            return nums[start];// 只有一个元素
        }

        vector<int> dp(nums.size());

        // 初始化dp
        dp[start] = nums[start];

        dp[start + 1] = max(nums[start],nums[start + 1]);// 从相邻两个屋子选取3最大的

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

        return dp[end];
    }

};