问题:
你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。
输入:[1,2,3,1]
输出:4
解释:偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
偷窃到的最高金额 = 1 + 3 = 4 。
第一种方法:
分析:
dp[n]的意义为第n间房所能偷窃的最大金额
由于不可以在相邻的房屋闯入,所以在当前位置 n 房屋可盗窃的最大值,要么就是 n-1 房屋可盗窃的最大值,要么就是 n-2 房屋可盗窃的最大值加上当前房屋的值,二者之间取最大值所以状态转移方程为:dp[n]=Max(dp[n-1],dp[n-2]+nums)
base case:dp[0]=0,dp[1]=nums[0];
代码:
Java语言
class Solution {
public int rob(int[] nums) {
int len = nums.length;
if(len == 0)
return 0;
int[] dp = new int[len + 1];
dp[0] = 0;
dp[1] = nums[0];
for(int i = 2; i <= len; i++) {
dp[i] = Math.max(dp[i-1], dp[i-2] + nums[i-1]);
}
return dp[len];
}
}
这里dp[i]中的i表示的就是第i间房间,而nums中,下标为i-1才表示第i间房的现金量
复杂度:
时间复杂度:O(n)
空间复杂度:O(n)
第二种方法:
状态定义:
设动态规划列表 dp ,dp[i] 代表前 i个房子在满足条件下的能偷窃到的最高金额,则第i个房子能偷窃到的最高金额为dp[i+1]
状态转移方程:
dp[n+1] = max(dp[n],dp[n-1]+num)
初始状态:
dp[0]=0
返回值:
返回 dp 列表最后一个元素值,即所有房间的最大偷窃价值
由于dp[n] 只与 dp[n-1]和 dp[n-2]有关系,因此我们可以设两个变量 cur和 pre 交替记录,将空间复杂度降到 O(1)
class Solution {
public int rob(int[] nums) {
int pre = 0, cur = 0, tmp;
for(int num : nums) {
tmp = cur;
cur = Math.max(pre + num, cur);
pre = tmp;
}
return cur;
}
}
复杂度分析:
时间复杂度:O(n)
空间复杂度:O(1)