题目来源:https://leetcode.cn/problems/house-robber-ii/description/
C++题解(来源代码随想录): 对于一个数组,成环的话主要有如下三种情况:(1)考虑不包含首尾元素;(2)考虑包含首元素,不包含尾元素;(3)考虑包含尾元素,不包含首元素。“考虑”包含尾元素,但不一定要选尾部元素!而情况二 和 情况三 都包含了情况一了,所以只考虑情况二和情况三就可以了。
class Solution {
public:
int rob(vector<int>& nums) {
int len = nums.size();
if(len == 1) return nums[0];
if(len == 2) return max(nums[0], nums[1]);
// 不偷最后一个房间
vector<int> dp1(len-1);
dp1[0] = nums[0];
dp1[1] = max(nums[0], nums[1]);
// 不偷第一个房间
vector<int> dp2(len-1);
dp2[0] = nums[1];
dp2[1] = max(nums[1], nums[2]);
for(int i = 2; i < len-1; i++) {
dp1[i] = max(dp1[i-1], dp1[i-2]+nums[i]);
dp2[i] = max(dp2[i-1], dp2[i-2]+nums[i+1]);
}
return max(dp1[len-2], dp2[len-2]);
}
};
class Solution {
public:
int rob(vector<int>& nums) {
int len = nums.size();
if(len == 1) return nums[0];
if(len == 2) return max(nums[0], nums[1]);
// 不偷最后一个房间
vector<int> dp1(len-1);
dp1[0] = nums[0];
dp1[1] = max(nums[0], nums[1]);
// 不偷第一个房间
vector<int> dp2(len-1);
dp2[0] = nums[1];
dp2[1] = max(nums[1], nums[2]);
for(int i = 2; i < len-1; i++) {
dp1[i] = max(dp1[i-1], dp1[i-2]+nums[i]);
dp2[i] = max(dp2[i-1], dp2[i-2]+nums[i+1]);
}
return max(dp1[len-2], dp2[len-2]);
}
};