LeetCode 213. 打家劫舍 II

现在是个环形,第一家和最后一家连着,所以和打家劫舍I不同了,主要思想是仍然看成两个数组(分成两种情况,最后比较一下),一个是0-length-2,一个是1-length-1。因为抢了0就不能抢最后一家,不抢0才能抢最后一家。

 public int rob(int[] nums) {
        if (nums.length == 0) {
            return 0;
        }
        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];
        dp1[0] = nums[0];
        dp1[1] = Math.max(nums[0], nums[1]);
        for (int i = 2; i < nums.length - 1; i++) {
            dp1[i] = Math.max(nums[i] + dp1[i - 2], dp1[i - 1]);
        }
        int[] dp2 = new int[nums.length];
        dp2[1] = nums[1];
        dp2[2] = Math.max(nums[1], nums[2]);
        for (int i = 3; i < nums.length; i++) {
            dp2[i] = Math.max(nums[i] + dp2[i - 2], dp2[i - 1]);
        }
        return Math.max(dp1[nums.length - 2], dp2[nums.length - 1]);
    }

 

你可能感兴趣的:(algorithm,LeetCode)