https://leetcode-cn.com/problems/house-robber-ii/submissions/
跟打家劫舍第一题相比就是把房子变成了环状。也就是第一个房子和最后一个房子不能一起偷了。因此可以把偷第1个到第n个房子的问题转换成两个分支,一个是计算偷盗第1个到第n-1个房子,一个是偷第2个到第n个房子。最后看这两个分支哪个钱多,就输出哪一个。注意这里还是用滚动数组来实现的动态规划,降低空间复杂度。
class Solution {
public:
int rob(vector<int>& nums) {
// 关键在于处理第一个房子和最后一个房子,如果没偷第一个房子,那就求第二个到最后一个房子总金额
// 如果偷了第一个房子,那就求第一个到倒数第二个总金额。然后总这两个之间找一个最大的
int len = nums.size();
if(len == 0) return 0;
if(len == 1) return nums[0];
if(len == 2) return max(nums[0], nums[1]);
int first_1 = nums[0];
int second_1 = max(nums[0], nums[1]);
int first_2 = nums[1];
int second_2 = max(nums[1], nums[2]);
for(int i = 2; i < len - 1; ++i){
int temp = second_1;
second_1 = max(first_1 + nums[i], second_1);
first_1 = temp;
}
for(int i = 3; i < len; ++i){
int temp = second_2;
second_2 = max(first_2 + nums[i], second_2);
first_2 = temp;
}
return max(second_1, second_2);
}
};