读题发现,数组每个元素的取舍都与其前一个元素和后一个元素的取舍相关,进一步可以分析,除了第一个元素之外,其余元素的取舍均和前一个元素的取舍相关,这引导我们往从头到尾遍历数组的方向上去解题,回过头来看题目的要求是要求最大值,因而我们可以尝试动态规划解法。
这里我个人尝试了两种状态迁移方程
##第一种
{
当1<=i<nums.size()
dp[i][1]=dp[i-1][0]+nums[i];
dp[i][0]=max(dp[i-1][0],dp[i-1][1]);
特殊情况
dp[0][0]=0;
dp[0][1]=nums[0];
}
//dp[i][j]指的是第i个住宅盗窃或者不盗窃情况下,小偷在前i个住宅所能获得的最大价值,其中j=0代表不盗窃这一个住宅而j=1代表盗窃该住宅
最终只需要选择dp[n-1][0]和dp[n-1][1]中较大者即可
代码如下:
class Solution {
public:
int rob(vector<int>& nums) {
int dp[2]={0,nums[0]};
int i=0;
for(i=1;i<nums.size();i++)
{
int t=dp[0];
dp[0]=dp[1]>dp[0]?dp[1]:dp[0];
dp[1]=t+nums[i];
}
return dp[0]>dp[1]?dp[0]:dp[1];
}
};
##第二种
{
当2<i<nums.size()
dp[i]=max{dp[i-2],dp[i-3]}+nums[i];
特殊情况
dp[0]=nums[0];
dp[1]=nums[1];
dp[2]=nums[0]+nums[2];
}//dp[i]代表前i个住宅最后偷窃的家庭为i所能获得的最大价值
代码如下:
class Solution {
public:
int rob(vector<int>& nums) {
int length=nums.size();
long long int dp[3];
int i;
dp[0]=nums[0];
if(length==1)
return dp[0];
dp[1]=nums[1];
if(length==2)
return nums[0]>nums[1]?nums[0]:nums[1];
dp[2]=nums[0]+nums[2];
for(i=3;i<nums.size();i++)
{
dp[0]=(dp[0]>dp[1]?dp[0]:dp[1])+nums[i];
if(++i>=nums.size())
break;
dp[1]=(dp[1]>dp[2]?dp[1]:dp[2])+nums[i];
if(++i>=nums.size())
break;
dp[2]=(dp[2]>dp[0]?dp[2]:dp[0])+nums[i];
}
length--;
return dp[length%3]>dp[(length-1)%3]?dp[length%3]:dp[(length-1)%3];
}
};
最后只需要比较dp[n-1]和dp[n-2]大小即可获得最大值
##第三种(官方解法)
{
当1<i<nums.size()
dp[i]=max{dp[i-1],dp[i-2]+nums[i]}];
特殊情况
dp[0]=nums[0];
dp[1]=nums[1];
}
//dp[i]指的是前i个住宅小偷所得偷的最大价值
dp[n-1]为所求值
代码暂无