这道题目 https://cloud.tencent.com/developer/article/1581788 一样,改进了一点,就是首尾也是相邻的。
那么我们在DP的时候,还要考虑第一个房子有没有被抢劫的情况。所以状态数组变成了DPik,j表示i个房子是否抢劫,k表示第1个房子是否抢劫。
代码语言:javascript复制class Solution {
public:
int dp[100005][2][2];
int rob(vector<int>& nums) {
if(nums.size()==0)
return 0;
if(nums.size()==1)
return nums[0];
dp[0][0][0]=0;
dp[0][1][1]=nums[0];
dp[0][0][1]=-1;
dp[0][1][0]=-1;
for(int i=1;i<nums.size();i )
{
if(dp[i-1][0][1]==-1)
dp[i][1][1]=-1;
else
dp[i][1][1] = dp[i-1][0][1] nums[i];
dp[i][1][0] = dp[i-1][0][0] nums[i];
dp[i][0][1] = max(dp[i-1][1][1],dp[i-1][0][1]);
dp[i][0][0] = max(dp[i-1][1][0],dp[i-1][0][0]);
}
return max(dp[nums.size()-1][1][0],max(dp[nums.size()-1][0][1],dp[nums.size()-1][0][0]));
}
};