Leetcode198/213 打家劫舍 题解 [动态规划]

题目描述

你是一个专业的小偷，计划偷窃沿街的房屋。每间房内都藏有一定的现金，影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组，计算你 不触动警报装置的情况下 ，一夜之内能够偷窃到的最高金额。

解法分析

这道题目显然是一道典型的动态规划题，通过题目已知的关系，相邻的两间房屋不能同时被偷，可以找到递推公式：
$$ans[n]=max(ans[n-1],(ans[n-2]+nums[n]))(n>2)$$

class Solution {
    public int rob(int[] nums) {
        int[] ans=new int[nums.length];
         if(nums.length==1) return nums[0];
         if(nums.length==2) return compare(nums[0],nums[1]);
        ans[0]=nums[0];
        ans[1]=compare(nums[0],nums[1]);
        for(int i=2;i<ans.length;i++){
            ans[i]=compare(ans[i-1],(ans[i-2]+nums[i]));
        }
        return ans[nums.length-1]; 
    }
    public int compare(int a,int b){
        if(a>b) return a;
        else return b;
    }
}

变式训练

如果将第一栋房屋和最后一栋房屋连起来呢？

变式题解

可以分成两类考虑，一类是去掉最后一栋房屋，第二类是去掉第一栋房屋

class Solution {
    public int rob(int[] nums) {
         if(nums.length==1) return nums[0];
         if(nums.length==2) return compare(nums[0],nums[1]);
        return compare(RangeRob(nums,0,nums.length-2), RangeRob(nums,1,nums.length-1));
    }
    public int compare(int a,int b){
        if(a>b) return a;
        else return b;
    }
    public int RangeRob(int[] nums,int begin,int end){
        int[] ans=new int[nums.length-1];
        ans[0]=nums[begin];
        ans[1]=compare(nums[begin],nums[begin+1]);
        for(int i=2;i<nums.length-1;i++){
             ans[i]=compare(ans[i-1],(ans[i-2]+nums[i+begin]));
        }
        return ans[nums.length-2];
    }
}