题目
解答
和之前的198.打家劫舍类似,但有个限制就是数组首尾相邻,因此分两种情况:从0开始和从1开始。
代码中dp1代表从0开始,dp2代表从1开始,均有如下状态转移方程:
f(n) = max{f(n-1), f(n-2)+nums(n)}
(f(n)代表到第n个房屋为止获得的最大金额)
class Solution {
public:
int rob(vector<int>& nums) {
if (nums.size() == 0) return 0;
if (nums.size() == 1)
return nums[0];
if (nums.size() == 2)
return max(nums[0], nums[1]);
if (nums.size() == 3)
return max(max(nums[0], nums[1]), nums[2]);
int dp1[2] = {0};
dp1[0] = nums[0];
dp1[1] = max(nums[0], nums[1]);
for (int i = 2; i < nums.size()-1; i++) {
int temp = dp1[1];
dp1[1] = max(dp1[0]+nums[i], dp1[1]);
dp1[0] = temp;
}
int dp2[2] = {0};
dp2[0] = nums[1];
dp2[1] = max(nums[1], nums[2]);
for (int i = 3; i < nums.size(); i++) {
int temp = dp2[1];
dp2[1] = max(dp2[0]+nums[i], dp2[1]);
dp2[0] = temp;
}
return max(dp1[1], dp2[1]);
}
};