213. 打家劫舍 II

题目

思路：动态规划

• 环形的话，第一个房子和最后一个房子，不能同时考虑。
• 把环拆成两个队列，一个是从0到n-1，另一个是从1到n，然后返回两个结果最大的。

代码

class Solution {
    public int rob(int[] nums) {
        if (nums.length == 1) {
            return nums[0];
        }
        if (nums.length == 2) {
            return Math.max(nums[0], nums[1]);
        }
        
        int[] dp1 = new int[nums.length - 1];
        int[] dp2 = new int[nums.length - 1];
        
        dp1[0] = nums[0];
        dp2[0] = nums[1];
        
        dp1[1] = Math.max(nums[0], nums[1]);
        dp2[1] = Math.max(nums[1], nums[2]);
        
        // 第一种偷法
        for (int i = 2; i <= nums.length - 2; i++) {
            //考虑了 2～length-2的nums
            dp1[i] = Math.max(dp1[i - 2] + nums[i], dp1[i - 1]); 
        }
        //第二种偷法
        for (int i = 2; i <= nums.length - 2; i++) {
            //考虑了 3～length-1的nums
            dp2[i] = Math.max(dp2[i - 2] + nums[i + 1], dp2[i - 1]); 
        }
        // 第一种偷法最大值
        int max1 = 0;
        for (int x : dp1) {
            max1 = Math.max(max1, x);
        }
        // 第二种偷法最大值
        int max2 = 0;
        for (int x : dp2) {
            max2 = Math.max(max2, x);
        }
        return Math.max(max1, max2);
    }
}