你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。
示例 1:
输入:[1,2,3,1]
输出:4
解释:偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
偷窃到的最高金额 = 1 + 3 = 4 。
示例 2:
输入:[2,7,9,3,1]
输出:12
解释:偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。
偷窃到的最高金额 = 2 + 9 + 1 = 12 。
提示:
0 <= nums.length <= 100
0 <= nums[i] <= 400
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/house-robber
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:动态规划。
假设有5
栋房子,编号为1、2、3、4、5
,考查最后一栋房子的情况,假如我偷这一栋房子,那么4
号房子就不能偷,假如我不偷5
号房子,那么4
号房子可偷也可不偷。我们用f[i]
来表示“前i
栋房子能够偷到的钱数的最大值”,那么可以得到状态转移方程为:
f[i] = max(f[i-1],f[i-2]+nums[i-1])
注意:这里的下标是从0
开始的。
其中f[i-1]
就表示第i
栋房子我没偷,f[i-2]+nums[i-1]
,就表示第i
栋房子我偷了。
边界条件:
f[0]=0
,前0
栋房子什么也偷不到。f[1] = nums[0]
,前1
栋房子必偷,f[2] = max(nums[0],nums[1])
,前两栋房子偷其中最大者。
代码实现:
class Solution {
public:
int rob(vector<int>& nums) {
int n = nums.size();
if(n == 0)
return 0;
if(n == 1)
return nums[0];
vector<int>f(n+1,0);
//初始条件
f[0] = 0;//前0栋房子什么也偷不到
f[1] = nums[0];//前1栋房子必偷
for(int i = 2;i <= n;++i)
{
f[i] = max(f[i-1],f[i-2]+nums[i-1]);
}
return f[n];
}
};
优化一下空间:
class Solution {
public:
int rob(vector<int>& nums) {
int n = nums.size();
if(n == 0)
return 0;
if(n == 1)
return nums[0];
int prev_prev = 0;//f[0]
int prev = nums[0];//f[1]
int cur = 0;
for(int i = 2;i <= n;++i)
{
//prev = f[i-1]
//prev_prev = f[i-2]
cur = max(prev,prev_prev + nums[i-1]);
prev_prev = prev;
prev = cur;
}
return cur;
}
};