原题地址:https://leetcode-cn.com/problems/house-robber-ii/description/
题目描述:
你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。
示例 1:
输入: [2,3,2] 输出: 3 解释: 你不能先偷窃 1 号房屋(金额 = 2),然后偷窃 3 号房屋(金额 = 2), 因为他们是相邻的。
示例 2:
输入: [1,2,3,1] 输出: 4 解释: 你可以先偷窃 1 号房屋(金额 = 1),然后偷窃 3 号房屋(金额 = 3)。 偷窃到的最高金额 = 1 + 3 = 4 。
解题方案:
动态规划的题型,数组其实是一个环形,主要思想是仍然看成两个数组(分成两种情况,最后比较一下),一个是0-length-2,一个是1-length-1。因为抢了0就不能抢最后一家,不抢0才能抢最后一家。
代码:
class Solution {
public:
int rob(vector<int>& nums) {
if (nums.size() == 0) return 0;
if (nums.size() == 1) return nums[0];
if (nums.size() == 2) return max(nums[0], nums[1]);
int dp1[nums.size() - 1];
dp1[0] = nums[0];
dp1[1] = max(nums[0], nums[1]);
for (int i = 2; i < nums.size() - 1; i++)
dp1[i] = max(nums[i] + dp1[i - 2], dp1[i - 1]);
int dp2[nums.size()];
dp2[1] = nums[1];
dp2[2] = max(nums[1], nums[2]);
for (int i = 3; i < nums.size(); i++)
dp2[i] = max(nums[i] + dp2[i - 2], dp2[i - 1]);
return max(dp1[nums.size() - 2], dp2[nums.size() - 1]);
}
};
领扣给出的最短时间代码:
class Solution {
public:
int rob(vector<int>& nums) {
if(nums.size() == 0)
return 0;
if(nums.size() == 1)
return nums[0];
if(nums.size() == 2)
return max(nums[0],nums[1]);
return max(process(nums,0,nums.size()-1),process(nums,1,nums.size()));
}
int process(vector<int>& nums,int start,int end){
vector<int> result(nums.size(),0);
result[start] = nums[start];
result[start+1] = max(nums[start],nums[start+1]);
for(int i = start+2; i < end; i++) {
result[i] = max(result[i-2] + nums[i] , result[i-1]);
}
return result[end - 1];
}
};