解题思路:若第一座房子和最后一座房子只抢其一,去掉不抢的那栋,此题就回归到198 House Robber 的解!
代码:
class Solution {
#define _MAX(a,b) ((a)>(b))?(a):(b)
public:
int firstRob(vector<int>& nums,int n) {//此函数取自198 House Robber,仅增加了参数列表中的int n
if(n==0)return 0;
if(n==1)return nums[0];
int d[n]={0};
d[0]=nums[0];
d[1]=_MAX(nums[0],nums[1]);
for(int i=2;i<n;++i){
d[i]=_MAX(d[i-2]+nums[i],d[i-1]);
}
return d[n-1];
}
public:
int rob(vector<int>& nums) {//动态规划,第一座房子和最后一座房子只能抢其一,然后删掉不抢的那栋,回到198 House Robber的解!
int n=nums.size();
if(n==0)return 0;
if(n==1)return nums[0];
int rob1=firstRob(nums,n-1);//不抢最后一座房子
nums.erase(nums.begin());
int rob2=firstRob(nums,n-1);//不抢第一座房子
return _MAX(rob1,rob2);
}
};