Leetcode 213.大家劫舍II

打家劫舍II

你是一个专业的小偷，计划偷窃沿街的房屋，每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈，这意味着第一个房屋和最后一个房屋是紧挨着的。同时，相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。

给定一个代表每个房屋存放金额的非负整数数组，计算你在不触动警报装置的情况下，能够偷窃到的最高金额。

示例 1:

输入: [2,3,2]

输出: 3

解释: 你不能先偷窃 1 号房屋（金额 = 2），然后偷窃 3 号房屋（金额 = 2）, 因为他们是相邻的。

示例 2:

输入: [1,2,3,1]

输出: 4

解释: 你可以先偷窃 1 号房屋（金额 = 1），然后偷窃 3 号房屋（金额 = 3）。

  偷窃到的最高金额 = 1 + 3 = 4 。

基本思路：

第一个房屋和最后一个房屋是紧挨着的，说明第一个房屋和最后一个房屋不能同时盗取。我们可以考虑两种情况：

（1）考虑偷取[0, n - 2]的房屋。

（2）考虑偷取[1, n - 1]的房屋。

取上述两种情况的大者即为答案。

class Solution {
    public int rob(int[] nums) {
        if(nums.length==0) return 0;
        if(nums.length==1) return nums[0];
        int[] leftNums=new int[nums.length-1];
        int[] rightNums=new int[nums.length-1];
        for(int i=0;i<nums.length-1;i++){
            leftNums[i]=nums[i];
        }
        for(int i=1;i<nums.length;i++){
            rightNums[i-1]=nums[i];
        }
        int left=kRob(leftNums);
        int right=kRob(rightNums);
        return left>right?left:right;
    }

    public int kRob(int[] nums) {
        int n=nums.length;
        if(n==0) return 0;
        if(n==1) return nums[0];
        int[] dp=new int[n];
        dp[0]=nums[0];
        dp[1]=nums[1];
        if(dp[1]<dp[0])dp[1]=dp[0];
        for(int i=2;i<n;i++){
            dp[i]=dp[i-2]+nums[i];
            if(dp[i]<dp[i-1]) dp[i]=dp[i-1];
        }
        return dp[n-1];
    }
}