原题链接:Leetcode 213. 打家劫舍 II
cpp
复制代码
class Solution {
public:
int rob(vector<int>& nums) {
int n = nums.size();
if (n == 1)
return nums[0];
if (n == 2)
return max(nums[0], nums[1]);
// 如果偷了第一家,就不能偷最后一家
int dp[n - 1];
dp[0] = nums[0];
dp[1] = nums[0];
for (int i = 2; i < n - 1; i++) {
dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);
}
// 如果从第二家开始偷,就没事
int dp2[n];
dp2[0] = 0;
dp2[1] = nums[1];
dp2[2] = max(nums[1], nums[2]);
for (int i = 3; i < n; i++) {
dp2[i] = max(dp2[i - 2] + nums[i], dp2[i - 1]);
}
return max(dp[n - 2], dp2[n - 1]);
}
};