动态规划
比剑指 Offer II 089. 房屋偷盗多了一点,就是我们偷第一家就不能偷最后一家,偷第二家可以偷最后一家,所以再这个基础之上分开讨论,就与那个题相同了。
从第一个开始偷,不偷最后一个
从第二个开始偷,可以偷最后一个
取两种结果的最大值返回即可
class Solution {
public:
int rob(vector<int>& nums) {
// 环形的,那么我们从第一个开始偷一遍,然后再从第二个开始偷一遍,然后取两者的最大值
if(nums.size()==0){
return 0;
}
if(nums.size()==1){
return nums.back();
}
if(nums.size()==2){
return max(nums[0],nums[1]);
}
// 从第一个开始偷,不偷最后一个
vector<int> res_1;
res_1.push_back(0);
res_1.push_back(nums[0]);
res_1.push_back(nums[1]);
for(int i=2;i<nums.size()-1;i++){
res_1.push_back(nums[i]+max(res_1[i-1],res_1[i-2]));
cout<<res_1.back()<<endl;
}
// 从第二个开始偷,偷最后一个,不偷第一个
vector<int> res_2;
res_2.push_back(0);
res_2.push_back(nums[1]);
res_2.push_back(nums[2]);
for(int i=3;i<nums.size();i++){
res_2.push_back(nums[i]+max(res_2[i-2],res_2[i-3]));
cout<<res_2.back()<<endl;
}
return max(max(res_1[res_1.size()-1],res_1[res_1.size()-2]),max(res_2[res_2.size()-1],res_2[res_2.size()-2]));
}
};