本题和 打家劫舍 的区别在于房屋围成了一圈,那么在选了第一个房子,就无法选择最后一个房子;选了最后一个房子,就无法选择第一个房子。
我们只需要执行两次 打家劫舍 就可以,一次以第一个房子为起点,倒数第二个房子为终点,一次以第二个房子为起点,最后一个房子为终点,最后选择两次结果的最大值即可
class Solution {
public:
int rob(vector<int>& nums) {
if(nums.size() == 1){
return nums[0];
}
int result1 = robonce(nums, 0, nums.size() - 1);
int result2 = robonce(nums, 1, nums.size());
return max(result1, result2);
}
int robonce(vector<int>& nums, int start, int end){
if(end - start == 1){
return nums[start];
}
vector<int> dp(nums.size(), 0);
dp[start] = nums[start];
dp[start + 1] = max(nums[start], nums[start + 1]);
for(int i = start + 2; i < end; i++){
dp[i] = max(dp[i - 1], dp[i - 2] + nums[i]);
}
return dp[end - 1];
}
};