问题描述 链接到标题
解题思路 链接到标题
参照198.打家劫舍,但是这里多了一个首尾不能同时选择的选项,因此可以考虑将数组分成两部分,一个包含[0, n - 1)
,一个包含[1, n)
,分别对应dp0
和dp1
,取最后两者的最大值即可。
代码 链接到标题
class Solution {
public:
int rob(vector<int>& nums) {
if (nums.size() == 0) return 0;
if (nums.size() == 1) return nums[0];
int result1 = robRange(nums, 0, nums.size() - 2);
int result2 = robRange(nums, 1, nums.size() - 1);
return max(result1, result2);
}
int robRange(vector<int>& nums, int start, int end) {
if (end == start) return nums[start];
vector<int> dp(nums.size());
dp[start] = nums[start];
dp[start + 1] = max(nums[start], nums[start + 1]);
for (int i = start + 2; i <= end; i++) {
dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);
}
return dp[end];
}
};