题目:
你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。
示例:
输入:[1,2,3,1]
输出:4
解释:偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
偷窃到的最高金额 = 1 + 3 = 4 。
输入:[2,7,9,3,1]
输出:12
解释:偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。
偷窃到的最高金额 = 2 + 9 + 1 = 12 。
nums
1 | 2 | 3 | 1 |
res:
1 | max(nums[0],nums[1]) | max(nums[2]+res[0],res[1]) | max(nums[3]+res[1],res[2]) |
上面为两个数组,nums数组为传入数组,res为结果数组,最后的结果为res中最后一个数组中存在的数据。
我们使用动态规划来做题,我们先来寻找动态规划的递推式。
当如果输入只有一个数据时,我们只需要返回该数据就可以,当数据为两个时,我们返回两者之间最大的数据,当数组的长度为 3时,这时,我们只需计算第一个和第三个相加与第二个之和相比较哪个最大,当数组长度为4时,我们需要计算当第二个最优解时加上最后一个最优解时的数据与前一个的结果相比较,选择最大的。
所以最终的递推式为:
res[i] = max(res[i-2]+nums[i], res[i-1])
提交代码:
public static int rob(int[] nums) {
int con = nums.length;
int[] res = new int[con];
if (con == 1)
return nums[0];
if (con ==2){
int a = Math.max(nums[0], nums[1]);
return a;
}
res[0] = nums[0];
res[1] = Math.max(nums[0],nums[1]);
for(int i =2;i<con;i++){
res[i] = Math.max(res[i-2]+nums[i], res[i-1]);
}
return res[con-1];
}
执行结果: