题意:
给定一个整数数组nums,数组中的元素表示每个房间存有的现金数额,你不能偷相邻的两家,并且第一个房间和最后一个房间视为相邻。
请你计算在不被发现的前提下最多的偷窃金额。
方法一:
动态规划
思路:dp[ i ]表示前i个房间最多的偷窃金额。
因为第一个房间和最后一个房间视为相邻,所以可以划分为两个区间【1:n-1】和【2:n】。最后取两者的最大值。
class Solution {
public:
int dp1[200005]={0};//dp[i]表示前i个房间最多的偷窃金额
int dp2[200005]={0};
int rob(vector<int>& nums) {
int n=nums.size();
if(n==1)
return nums[0];
dp1[1]=nums[0];//初始化
for(int i=2;i<=n-1;i++){//区间[1:n-1]
dp1[i]=max(dp1[i-1],dp1[i-2]+nums[i-1]);//状态转移方程
}
for(int i=2;i<=n;i++){//区间[2:n]
dp2[i]=max(dp2[i-1],dp2[i-2]+nums[i-1]);//状态转移方程
}
return max(dp1[n-1],dp2[n]);
}
};
时间复杂度:
空间复杂度:![]()
方法二:
滚动数组优化
思路:根据转移转移方程,可知当前项只与前两项有关。
因此可以不用一维数组,改用几个变量实现。(滚动数组)
class Solution {
public:
int rob(vector<int>& nums) {
int n=nums.size();
if(n==1)
return nums[0];
if(n==2)
return max(nums[0],nums[1]);
int t1=nums[0],t2=max(nums[0],nums[1]);//初始化
for(int i=2;i<=n-2;i++){//区间[0:n-2]
int t=max(t1+nums[i],t2);//状态转移方程
t1=t2;
t2=t;
}
int t3=nums[1],t4=max(nums[1],nums[2]);//初始化
for(int i=3;i<=n-1;i++){//区间[1:n-1]
int t=max(t3+nums[i],t4);//状态转移方程
t3=t4;
t4=t;
}
return max(t2,t4);
}
};
时间复杂度:
空间复杂度:![]()



京公网安备 11010502036488号