题目
这不是今天的每日一题。这是前几天和男票随机摇了一题比谁做的快。(那天我俩都没做出来就搁置了)。今天我俩又捡起来看,我在他之前搞出来了。
https://leetcode-cn.com/problems/house-robber-ii/
答案
我使用的是双层dp。
最大值沿用dp思路,如果最大值包含最后一个值和最大值不包含本身。
最大值不包含最后一个值:正常dp的dp[length-1]
最大值包含最后一个值:去掉首尾和倒数第二个值的dp[newLength] + 最后一个值
- 先忽略头尾不能并存的规则,按照普通思路求出dp数组。
dp[i] = Math.max(dp[i-1], dp[i-2] + nums[i])
- 如果length小于4可以直接return 结果。如果大于等于4,继续往下走。
- copy一个数组,长度是length-3,就是去掉第一个,和最后两个。求出新的dp。
- 结果就是两者取大值。
class Solution {
public int rob(int[] nums) {
int length = nums.length;
if (length == 0) {
return 0;
}
int[] dp = new int[length + 1];
int res = format(length, dp, nums);
if (res != -1) {
return res;
}
int[] newNums = new int[nums.length - 3];
System.arraycopy(nums, 1, newNums, 0, nums.length - 3);
int[] newDp = new int[newNums.length + 1];
format(newNums.length, newDp, newNums);
return Math.max(dp[length-1], newDp[newNums.length] + nums[length - 1]);
}
private int format(int length, int[] dp, int[] nums) {
dp[1] = nums[0];
for (int i = 2; i <= length; i++) {
dp[i] = Math.max(dp[i - 1], dp[i-2] + nums[i-1]);
}
if (length < 4) {
if (length < 3) {
return dp[length];
} else {
Arrays.sort(nums);
return nums[2];
}
}
return -1;
}
public static void main(String[] args) {
int[] nums = new int[]{4,1,2};
System.out.println(new Solution().rob(nums));
}
}
总结
值得骄傲的是,时间是一样的0ms,但是我使用的内存比官方题解少了0.1MB,提高了8%。
想了挺久的,之前思路有问题,还以为很简单。写错的时候一直在打补丁,以为是自己没有考虑全面,后面觉得不对,就把前面代码全删掉重新来了。