参考链接
- https://leetcode-cn.com/problems/house-robber-ii/
题目描述
你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。
解题思路
题目与打家劫舍1差不多,主要区别在于,这些房子不是一排,而是围成了一个圈。即第一间房子和最后一间房子也是相邻的,不能同时取钱。
显然,特殊的情况特殊考虑就行了。分两种情况,一种是只能取第一间房子到倒数第二间房子的钱,另一种情况是只能取第二间房子到最后一间房子的钱。而这两种情况,可以直接当成打家劫舍1来处理。
代码
略微修改一下打家劫舍1的代码即可。
class Solution {
public:
int robRange(vector<int> &nums, int start, int end)
{
int dp_i_1 = 0;
int dp_i_2 = 0;
int dp_i = 0;
for (int i = end; i >= start; i --)
{
dp_i = max(dp_i_1, dp_i_2 + nums[i]);
dp_i_2 = dp_i_1;
dp_i_1 = dp_i;
}
return dp_i;
}
int rob(vector<int>& nums) {
int n = nums.size();
if (n == 1)
{
return nums[0];
}
return max(robRange(nums, 0, n-2), robRange(nums, 1, n-1));
}
};