你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都 围成一圈 ,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警 。
给定一个代表每个房屋存放金额的非负整数数组,计算你 在不触动警报装置的情况下 ,能够偷窃到的最高金额,示例如下:
因为可能会成环,所以需要考虑三种情况:
(1)不包含收尾的最大值
(2)只包含首,不包含尾的最大值
(3)只包含尾,不包含首的最大值
后两种情况包含第一种情况,所以只需要讨论后两种情况即可。
动态规划五部曲:
(1)dp数组及下标含义:dp[j]为前j个房屋能偷到的最高金额;
(2)确定递推公式:dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);
(3)dp数组初始化:dp[0] = nums[0] dp[1] = max(nums[0], nums[1])
(4)遍历顺序:从前到后遍历
(5)举例推导dp数组
实现代码如下:
int rob(vector<int>& nums) {
if(nums.size() == 0) return 0;
else if(nums.size() == 1) return nums[0];
int result1 = robRange(nums,0,nums.size()-2); //情况II
int result2 = robRange(nums,1,nums.size()-1); //情况III
return max(result1, result2);
}
//每个单独情况
int robRange(vector<int>& nums, int start, int end) {
if(end == start) return nums[start];
//1.dp数组及下标含义
//dp[j]为前j个房屋能偷到的最高金额
//3.dp数组初始化
vector<int> dp(nums.size(), 0);
dp[start] = nums[start];
dp[start+1] = max(nums[start], nums[start + 1]);
//2.递推公式
//4.遍历顺序
for(int i = start + 2; i<=end; i++) {
dp[i] = max(dp[i-1], dp[i-2] + nums[i]);
}
return dp[end];
//5.举例推导dp数组
}