LeetCode刷题——213. 打家劫舍 II

面试刷题第13天

LeetCode刷题——213. 打家劫舍 II

1、题目描述

你是一个专业的小偷，计划偷窃沿街的房屋，每间房内都藏有一定的现金。这个地方所有的房屋都 围成一圈 ，这意味着第一个房屋和最后一个房屋是紧挨着的。同时，相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警 。
给定一个代表每个房屋存放金额的非负整数数组，计算你在不触动警报装置的情况下 ，今晚能够偷窃到的最高金额。

2、解题思路

由于首尾是相连通的，分两种情况：
（1）第一，在1到n-1范围上求；
（2）第二，在0到n-2范围上求；
（3）最后，取两种情况下的最大值。
Java代码如下：

class Solution {
    public int rob(int[] nums) {
        int n = nums.length;
        if(n == 1){
            return nums[0];
        }
        if(n == 2){
            return Math.max(nums[0], nums[1]);
        }
        //第一种情况：0到n-2上求
        int p1 = nums[0];
        int p2 = Math.max(nums[0], nums[1]);
        for(int i=2; i<n-1; i++){
            int temp = Math.max(p2, nums[i]+p1);
            p1 = p2;
            p2 = temp;
        }
        int ans1 = p2;
        //第二种情况：1到n-1上求
        p1 = nums[1];
        p2 = Math.max(nums[1], nums[2]);
        for(int i=3; i<n; i++){
            int temp = Math.max(p2, nums[i]+p1);
            p1 = p2;
            p2 = temp;
        }
        int ans2 = p2;
        return Math.max(ans1, ans2);
    }
}