- 这道题就是对打家劫舍1的扩充版本, 分成两种情况来讨论
- 这个主要依赖之前的两个状态
- 实现细节上可以使用同一个dp数组,保证最初的2个状态被设定好就可以
- 对于数组的处理要小心
- 当使用到nums[2]的时候, 就要意识到数组至少要有3个数字
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]);
}
int n = nums.size();
vector<int> dp(n, 0);
dp[0] = nums[0];
dp[1] = max(nums[0], nums[1]);
for (int i = 2; i < n - 1; i++) {
dp[i] = max(dp[i-2] + nums[i], dp[i-1]);
}
int result1 = dp[n-2];
dp[1] = nums[1];
dp[2] = max(nums[1], nums[2]);
for (int i = 3; i < n; i++) {
dp[i] = max(dp[i-2] + nums[i], dp[i-1]);
}
int result2 = dp[n-1];
return max(result1, result2);
}
};