leetcode每日一题213. 打家劫舍 II20210803

题目

这不是今天的每日一题。这是前几天和男票随机摇了一题比谁做的快。（那天我俩都没做出来就搁置了）。今天我俩又捡起来看，我在他之前搞出来了。

https://leetcode-cn.com/problems/house-robber-ii/

答案

我使用的是双层dp。
最大值沿用dp思路，如果最大值包含最后一个值和最大值不包含本身。
最大值不包含最后一个值：正常dp的dp[length-1]
最大值包含最后一个值：去掉首尾和倒数第二个值的dp[newLength] + 最后一个值

1. 先忽略头尾不能并存的规则，按照普通思路求出dp数组。

dp[i] = Math.max(dp[i-1], dp[i-2] + nums[i])

2. 如果length小于4可以直接return 结果。如果大于等于4，继续往下走。
3. copy一个数组，长度是length-3，就是去掉第一个，和最后两个。求出新的dp。
4. 结果就是两者取大值。

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%。
想了挺久的，之前思路有问题，还以为很简单。写错的时候一直在打补丁，以为是自己没有考虑全面，后面觉得不对，就把前面代码全删掉重新来了。