198. House Robber
你是一名职业强盗,计划沿街抢劫房屋。
每个房子都藏有一定数量的钱,阻止你抢劫每个房子的唯一限制是相邻的房子有连接的安全系统,如果两个相邻的房子在同一晚被闯入,它会自动联系警察。
给定一个整数数组 nums,代表每个房子的金额,返回今晚在不报警的情况下可以抢劫的最大金额。
EX
Example 1:
Input: nums = [1,2,3,1]
Output: 4
Explanation: Rob house 1 (money = 1) and then rob house 3 (money = 3).
Total amount you can rob = 1 + 3 = 4.
Example 2:
Input: nums = [2,7,9,3,1]
Output: 12
Explanation: Rob house 1 (money = 2), rob house 3 (money = 9) and rob house 5 (money = 1).
Total amount you can rob = 2 + 9 + 1 = 12.
Constraints:
1 <= nums.length <= 100
0 <= nums[i] <= 400
DP 的算法怎么解?
-
查找递归关系
-
递归(自上而下)
-
递归+mem(自上而下)
-
迭代+mem(自下而上)
-
迭代 + N 个变量(自下而上)
核心
所有的 dp 问题,用递归的时候,最核心的问题就是找到递归关系。
用迭代的时候,就是寻找递推公式。
V1-递归
递归关系
相邻的房子被偷盗,就会导致报警。
我们面对一个屋子 i,可以选择偷盗、或者不偷盗。
1)偷盗的话,
价值 = 当前屋子的偷盗价值 + 上一个屋子偷盗的价值 (i-2)
不能相邻,所以要看 i-2
2)不偷盗的话,当前获取为 0
价值 = 上一个屋子的偷盗价值(i-1)
可以相邻,看 i-1
选择最大价值:就是 2 个选择中的最大值作为结果。
递归终止条件:i < 0,返回 0。
java 实现
public int rob(int[] nums) {
return recursive(nums, nums.length-1);
}
private int recursive(int[] nums, int i) {
if(i < 0) {
return 0;
}
int robCur = recursive(nums, i-2) + nums[i];
int notRobCur = recursive(nums, i-1);
return Math.max(robCur, notRobCur);
}
当然,这种算法会在 55 / 70 超时。
因为一直再重复计算。
V2-递归+mem
思路
我们引入一个 mem 内存,缓存已经计算的结果,避免重复计算。
实现
public int rob(int[] nums) {
Integer[] mem = new Integer[nums.length];
return recursive(nums, nums.length-1, mem);
}
private int recursive(int[] nums, int i, Integer[] mem) {
int result = 0;
if(i < 0) {
return result;
}
// 获取 cache
if(mem[i] != null) {
return mem[i];
}
int robCur = recursive(nums, i-2, mem) + nums[i];
int notRobCur = recursive(nums, i-1, mem);
result = Math.max(robCur, notRobCur);
// cache
mem[i] = result;
return result;
}
耗时 0ms,超越 100%。
递归+mem 其实性能还是比较好的。
我们来看一下 dp 的方式,一般而言 dp 更加优雅,而且可以优化一下内存。
V3-dp
思路
我们首先把递归改成迭代。
整体变动其实不大,mem 数组改为 dp 数组。
dp[0] = 0;
dp[1] = nums[0];
递推公式:
我们从 i=1,开始遍历整个数组。
dp[i+1] = Math.max(dp[i], dp[i-1], nums[i]);
本质是一样的,就是两种选择中的最大值。
实现
public int rob(int[] nums) {
int[] dp = new int[nums.length + 1];
dp[0] = 0;
dp[1] = nums[0];
// 从1开始,避免 i-1 越界。
for(int i = 1; i < nums.length; i++) {
dp[i+1] = Math.max(dp[i], dp[i-1] + nums[i]);
}
return dp[nums.length];
}
V3-dp+内存优化
思路
当然,如果我们再追求极致一些,dp 数组都是可以优化的。
递推公式 dp[i+1] = Math.max(dp[i], dp[i-1] + nums[i]);
中,实际上只会依赖 dp[i-1]、dp[i]。
所以使用 2 个变量就可以搞定。
实现
public int rob(int[] nums) {
int pre1 = 0;
int pre2 = 0;
for(int val : nums) {
int temp = pre1;
pre1 = Math.max(pre1, pre2+val);
pre2 = temp;
}
return pre1;
}
213. 打家劫舍 II
你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。
这个地方所有的房屋都围成一圈,这意味着第一个房屋和最后一个房屋是紧挨着的。
同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警 。
给定一个代表每个房屋存放金额的非负整数数组,计算你 在不触动警报装置的情况下 ,今晚能够偷窃到的最高金额。
例子
示例 1:
输入:nums = [2,3,2]
输出:3
解释:你不能先偷窃 1 号房屋(金额 = 2),然后偷窃 3 号房屋(金额 = 2), 因为他们是相邻的。
示例 2:
输入:nums = [1,2,3,1]
输出:4
解释:你可以先偷窃 1 号房屋(金额 = 1),然后偷窃 3 号房屋(金额 = 3)。
偷窃到的最高金额 = 1 + 3 = 4 。
示例 3:
输入:nums = [1,2,3]
输出:3
提示:
1 <= nums.length <= 100
0 <= nums[i] <= 1000
V1-复用以前的算法
思路
这一题和前面的区别就在于屋子是环形排列的,首尾相连。
那么,应该如何解决呢?
如果有 10 个房子,a0, a1, …, a9
那么有两种偷取的方式:
1)偷取 a0
a1、a9 和 a0 都紧挨着。
那么子问题变成如何在 a2, …, a8 中获取最大收益。
2)不偷取 a0
那么子问题变成如何在 a1, …, a9 中获取最大收益。
然后我们取两种场景的最大值即可。
实现
public class T213_HouseRobberII {
public int rob(int[] nums) {
//rob0,从 2...-1 开始
int sum1 = nums[0] + robNoCircle(getSubArray(nums, 2, nums.length-2));
//not rob0
int sum2 = robNoCircle(getSubArray(nums, 1, nums.length-1));
return Math.max(sum1, sum2);
}
private int[] getSubArray(int[] nums,
int startIndex,
int endIndex) {
if(endIndex < startIndex) {
return new int[0];
}
int len = endIndex - startIndex + 1;
int[] results = new int[len];
int size = 0;
for(int i = startIndex; i <= endIndex; i++) {
results[size++] = nums[i];
}
return results;
}
// 无环时
public int robNoCircle(int[] nums) {
if(nums.length == 0) {
return 0;
}
int[] dp = new int[nums.length + 1];
dp[0] = 0;
dp[1] = nums[0];
// 从1开始,避免 i-1 越界。
for(int i = 1; i < nums.length; i++) {
dp[i+1] = Math.max(dp[i], dp[i-1] + nums[i]);
}
return dp[nums.length];
}
}
参考资料
https://leetcode.com/problems/house-robber/description/
https://leetcode.com/problems/house-robber/solutions/156523/from-good-to-great-how-to-approach-most-of-dp-problems/
https://leetcode.cn/problems/house-robber-ii/
https://leetcode.com/problems/house-robber-ii/solutions/893957/python-just-use-house-robber-twice/?orderBy=most_votes