（JAVA）LeetCode 198 打家劫舍

题目描述

• 你是一个专业的小偷，计划偷窃沿街的房屋。每间房内都藏有一定的现金，影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。

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

• LeetCode解释： 如果房屋数量大于两间，应该如何计算能够偷窃到的最高总金额呢？对于第 k(k>2) 间房屋，有两个选项：

  • 偷窃第 k 间房屋，那么就不能偷窃第 k−1 间房屋，偷窃总金额为前 k−2 间房屋的最高总金额与第 k 间房屋的金额之和。

  • 不偷窃第 k 间房屋，偷窃总金额为前 k−1 间房屋的最高总金额。

  • 在两个选项中选择偷窃总金额较大的选项，该选项对应的偷窃总金额即为前 k 间房屋能偷窃到的最高总金额。

解法：动态规划

• 根据上面的状态转移方程和边界，代码如下

class Solution {
    public int rob(int[] nums) {
        if(nums==null||nums.length==0) 
            return 0;
        if(nums.length==1)
            return nums[0];
        int dp[] = new int[nums.length];
        dp[0]=nums[0];
        dp[1]=Math.max(nums[0],nums[1]);
        for(int i=2;i<nums.length;i++){
            dp[i]=Math.max(dp[i-1],dp[i-2]+nums[i]);
        }
        return dp[nums.length-1];
    }
}

• 时间复杂度：O(n）
• 空间复杂度：O(1）

改进(滚动数组)

• 时间复杂度：O(n)
• 空间复杂度：O(1)
• 可以发现上述方法使用了数组存储结果，我们可以仅定义两个变量，进行动态滚动进行操作(因为观察动态转移方程，发现每次仅需n-1的最大和n-2的最大两个数据)。

class Solution {
    public int rob(int[] nums) {
        if(nums==null||nums.length==0) 
            return 0;
        if(nums.length==1)
            return nums[0];
        int first=nums[0];
        int second=Math.max(nums[0],nums[1]);
        for(int i=2;i<nums.length;i++){
            int temp = second;
            second = Math.max(second,first+nums[i]);
            first = temp;
        }
        return second;
    }
}