一个专业的小偷,计划偷窃一个环形街道上沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都 围成一圈
,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组 nums ,请计算 在不触动警报装置的情况下 ,今晚能够偷窃到的最高金额。
示例 1: 输入:nums = [2,3,2] 输出:3 解释:你不能先偷窃 1 号房屋(金额 = 2),然后偷窃 3 号房屋(金额 = 2), 因为他们是相邻的。
示例 2: 输入:nums = [1,2,3,1] 输出:4 解释:你可以先偷窃 1 号房屋(金额 = 1),然后偷窃 3 号房屋(金额 =3)。
偷窃到的最高金额 = 1 + 3 = 4 。
https://leetcode-cn.com/problems/PzWKhm/
与之前打家劫舍 I 多了 一个限制条件罢了。
打家劫舍1
与之前打家劫舍 I 多了 一个限制条件。即 不能同时偷第一间和最后一间房子。
那么就分别 去掉第一间dp和最后一间dp , 返回最大的即为最终结果。
class Solution {
public:
int rob(vector<int>& nums) {
if(nums.empty()) return 0;
int n=nums.size();
if(n==1){
return nums[0];
}
int dp1[105]={0},dp2[105]={0};
for(int i=1;i<n;i++){
dp1[i]=max(dp1[(i+n-1)%n],dp1[(n+i-2)%n]+nums[i]);
}
for(int i=0;i<n-1;i++){
dp2[i]=max(dp2[(i+n-1)%n],dp2[(n+i-2)%n]+nums[i]);
}
return max(dp1[n-1],dp2[n-2]);
}
};
写了两个for太难看了,改成一个函数就好看多了
其实我就是想写复杂点
class Solution {
public:
int rob_dp(int dp[],vector<int>& nums,int start,int end,int n){
for(int i=start;i<end;i++){
dp[i]=max(dp[(i+n-1)%n],dp[(n+i-2)%n]+nums[i]);
}
return dp[end-1];
}
int rob(vector<int>& nums) {
if(nums.empty()) return 0;
int n=nums.size();
if(n==1){
return nums[0];
}
int dp1[105]={0},dp2[105]={0};
return max(rob_dp(dp1,nums,0,n-1,n),rob_dp(dp2,nums,1,n,n));
}
};