LeetCode第 198 题:打家劫舍(C++)_qq_32523711的博客-CSDN博客的扩展。
数组首尾是一个圈,所以首尾也是相邻的。第一个想法是先别管是不是圈,反正就是第一个和最后一个不能同时抢。
那我们把序列分为0 ~ n-1,1 ~ n两部分,分别计算,最后返回最大值就可以了。
class Solution {
public:
int rob(vector<int>& nums) {
if(nums.size() == 0) return 0;
if(nums.size() == 1) return nums[0];
int pre = 0, cur = 0;
for(int i = 0; i < nums.size()-1; ++i){//0~n-1
int tmp = cur;
cur = max(cur, pre+nums[i]);
pre = tmp;
}
int res1 = cur;
pre = 0, cur = 0;
for(int i = 1; i < nums.size(); ++i){//1~n
int tmp = cur;
cur = max(cur, pre+nums[i]);
pre = tmp;
}
return max(cur, res1);
}
};