问题描述:
思路描述:
典型的动态规划题,之前做过打家劫舍 I,这道题和 I 唯一的不同是房子是首尾相邻的,意思偷了第一家就不能偷最后一家,反之亦然;解决思路和打家劫舍 I也相同,维护两个变量,一个记录上一家不偷的最大收益,一个记录上一家偷的最大收益;
* 如果上一家偷了,那么这一家就不能再偷,当前的最大收益就是上一家偷了的基础上 + 0
* 如果上一家没有偷,那么这一家就可以选择偷,当前最大收益也就是 max(上一家没偷的最大收益, 上一家偷了的最大收益) + 当前这家的财产。
因为题目要求首尾住户是相邻的,于是两次动态规划,第一次传入nums[0]为0的数组(第一家不偷),第二次传入nums[nums.size()-1] 为0的数组(第一家偷),比较两次谁大就返回谁。或者也可以分别传入nums数组的 [0:num.size()-1] 和 [1:nums.size()] 来模拟第一家偷或者不偷的情况,道理都是一样。
代码实现:
class Solution {
public:
int returnValue(vector<int>& nums){
int lastNotThief = 0, lastDoThief = nums[0];
for(int i=1; i<nums.size(); i++){
int tempLastNotThief = lastNotThief;
lastNotThief = max(lastNotThief, lastDoThief) + 0;
lastDoThief = tempLastNotThief + nums[i];
}
return lastDoThief>lastNotThief?lastDoThief:lastNotThief;
}
int rob(vector<int>& nums) {
int len = nums.size();
if(len == 1) return nums[0];
int temp = nums[0];
nums[0] = 0;
int v1 = returnValue(nums);
nums[0] = temp;
nums[nums.size()-1] = 0;
int v2 = returnValue(nums);
return v1>v2?v1:v2;
}
};