House Robber
问题描述:你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。
问题解决:这种题一开始还是想用递归来做,设置一个函数which(nums,n,i),其中nums代表每个房屋存放的金额,n代表房屋的数目,i表示当前偷到了第i个房屋。如果i = n-1,代表小偷已经偷到了最后一家,返回最后一家的金额。如果i >= n,则表示小偷最后一次偷盗是倒数第2家,之后不会再偷盗了,返回零。如果i < n-1,那么小偷能够偷到的最大金额为max(nums[i] + which(nums,n,i+2), nums[i+1] + which(nums,n,i+3))。
class Solution {
public:
int rob(vector<int>& nums) {
int len=nums.size();
if(len==0) return 0;
else if(len==1) return nums[0];
return which(nums,len,0);
}
int which(vector<int>& nums,int n,int i)
{
if(i>=n) return 0;
else if(i==n-1) return nums[i];
return max(nums[i] + which(nums,n,i+2), nums[i+1] + which(nums,n,i+3));
}
};
很遗憾,还是超时了。考虑用动态规划来做,我们设置一个一维数组dp,其中dp[i]表示在nums数组到第i个位置时能够偷到的最大金额。比如说nums={1,2,1,2,3},那么dp[0]为1,根据规则dp[1]只能在0和1这两个位置选择一个,则dp[1]=2,dp[3]在0,1和2三个位置中选择,根据规则小偷不能偷相邻的房屋,那么只有两种选择:只选择位置1或者选择位置0和2,我们要怎么选择呢?其实也很简单,我们只要选择那个得到更大金额的即可,即dp[i]=max(dp[i-1],dp[i-2]+nums[i])。
class Solution {
public:
int rob(vector<int>& nums) {
int len=nums.size();
if(len<1) return 0;
else if(len==1) return nums[0];
vector<int> dp;
dp.push_back(nums[0]);
dp.push_back(max(nums[0],nums[1]));
for(int i=2;i<len;i++)
{
dp.push_back(max(dp[i-1],dp[i-2]+nums[i]));
}
return dp.back();
}
};
beats 100.00 % of cpp submissions
根据上面的启发:dp[i]=max(dp[i-1],dp[i-2]+nums[i]),其实我们每一次只需要保存类似dp[i-1]和dp[i-2]这两个值即可,
class Solution {
public:
int rob(vector<int> nums){
int len=nums.size();
if(len<1) return 0;
else if(len==1) return nums[0];
int dp,dp1=0,dp2=0;
for(int i=0;i<len;i++)
{
dp=max(dp2,dp1+nums[i]);
dp1=dp2;
dp2=dp;
}
return dp;
}
};