题目链接:https://leetcode.com/problems/house-robber-ii/
你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。
示例 1:
输入: [2,3,2]
输出: 3
解释: 你不能先偷窃 1 号房屋(金额 = 2),然后偷窃 3 号房屋(金额 = 2), 因为他们是相邻的。
示例 2:
输入: [1,2,3,1]
输出: 4
解释: 你可以先偷窃 1 号房屋(金额 = 1),然后偷窃 3 号房屋(金额 = 3)。
偷窃到的最高金额 = 1 + 3 = 4 。
思路:
用两个dp数组来保存,第一个数组可以保存 0~nums.length-2的最大值。
第二个数组来计算1~nums.length-1的最大值,最终结果可以取两者的最大值。
这里的dp表达式为
dp[i]=Max(dp[i-1],dp[i-2]+nums[i]);
dp[i]表示截止到下标i时的最大偷窃金额(包括i)。
具体实现时,有一些下标的坑要注意一下,先看代码,然后仔细分析
Runtime: 0 ms, faster than 100.00% of Java online submissions for House Robber II.
Memory Usage: 34.2 MB, less than 100.00% of Java online submissions for House Robber II.
class Solution {
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]);
if(nums.length==3)
return Math.max(Math.max(nums[0],nums[1]),nums[2]);
int[] dpF=new int[nums.length];
int[] dpS=new int[nums.length];
dpF[0]=nums[0];dpF[1]=Math.max(nums[0],nums[1]);//特别容易犯错,这里的dpF[1]一定要初始化为两者最大的那个值
dpS[0]=0;dpS[1]=nums[1];//因为这里不计算nums[0]的值,所以初始化为0以免影响结果。
for(int i=2;i<nums.length;i++){
dpF[i]=Math.max(dpF[i-1],dpF[i-2]+nums[i]);
dpS[i]=Math.max(dpS[i-1],dpS[i-2]+nums[i]);
}
return Math.max(dpF[nums.length-2],dpS[nums.length-1]);
}
}
比如数组 4,1,2,7,5,3,1,对应关系如下:
index 0 1 2 3 4 5 6
nums 4 1 2 7 5 3 1
dpF 4 4 6 11 11 14 14*
dpS 0* 1 2 8 8 11 11
其中数字带右上标✳的表示这个数不应该列入最后的计算中。