问题:你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。
算法:
创建 memo 数组并初始化 -1 ,代表没有被计算过。
进入 dfs 函数,递归边界条件为 i<0 ,即没有可以被计算的房子了,return 。如果这个房子没有被计算过,此时有两种选择,选这个房子或者不选,return 较大的一种。如果这个房子已经被计算过了,就不必递归,可以直接返回本身已经被计算过的值。
代码:
class Solution {
public:
int rob(vector<int>& nums) {
int n = nums.size();
vector<int> memo(n,-1);// -1代表没有计算过
auto dfs = [&](auto &&dfs,int i) -> int{
if (i < 0) return 0;// 没有房子可选了 递归边界
if(memo[i] != -1) return memo[i];//这个房子被计算过
return memo[i] = max(dfs(dfs,i - 1),dfs(dfs,i - 2) + nums[i]);
};
return dfs(dfs,n - 1);// 从最后一个房子开始
}
};
改写成递推:
由上图可知,变换完成后数组要向后拓展两个位置。
代码:
class Solution {
public:
int rob(vector<int>& nums) {
int n = nums.size();
vector<int> f(n + 2);
for(int i = 0;i < n;i++) f[i + 2] = max(f[i + 1],f[i] + nums[i]);
return f[n+1];
}
};