你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。
示例 1:
输入: [1,2,3,1]
输出: 4
解释: 偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
偷窃到的最高金额 = 1 + 3 = 4 。
示例 2:
输入: [2,7,9,3,1]
输出: 12
解释: 偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。
偷窃到的最高金额 = 2 + 9 + 1 = 12 。
解题思路:Y表示偷这家,N表示不偷,以[2,7,3,9,1] 为例。
2 7 3 9 1
Y 2 7 5 16 8
N 0 2 7 7 16
由上面的结果我们可以看到一些规律,从第二家开始 本次要偷的时候得到的总金额为上次没偷时候的总金额+本次要偷的总金额。
本次不偷的时候得到的总金额为max(上次偷的时候的总金额,上次不偷的时候的总金额)。
经过上面的分析我们就可以写出程序
时间复杂度:O(n)
空间复杂度:O(n)
public class test0509 {
public static void main(String[] args) {
Solution s = new Solution();
int[] nums = {2,7,3,9,1};
int a = s.rob(nums);
System.out.println(a);
}
}
class Solution {
public int rob(int[] nums) {
if(nums.length == 0){
return 0;
}
int[][] arr = new int[nums.length][2];
arr[0][0] = nums[0];
for(int i = 1;i<nums.length;i++){
//偷第i家
arr[i][0] = arr[i-1][1] + nums[i];
//不偷第i家
arr[i][1] = Math.max(arr[i-1][0],arr[i-1][1]);
}
return arr[nums.length-1][0] > arr[nums.length-1][1] ?
arr[nums.length-1][0] : arr[nums.length-1][1];
}
}
这个代码写出来之后,突然发现可以优化,因为存入到数组之后,最后只使用数组的最后一行,那么我们就可以不用数组,用两个变量就可以了。本次优化时间复杂度不变,空间复杂度变小。
时间复杂度:O(n)
空间复杂度:O(1)
public class test0509 {
public static void main(String[] args) {
Solution s = new Solution();
int[] nums = {2,7,3,9,1};
int a = s.rob(nums);
System.out.println(a);
}
}
class Solution {
public int rob(int[] nums) {
if(nums.length == 0){
return 0;
}
int N = 0;
int Y = nums[0];
for(int i = 1;i<nums.length;i++){
int tmp = Y;
Y = N+nums[i];
N = tmp > N ? tmp : N;
}
return Y > N ? Y : N;
}
}