用dp[i]表示前i间房屋能偷窃到的最高总金额,那么就有如下的状态转移方程
dp[i]=max(dp[i-2]+num[i],dp[i-1])
边界条件
dp[0] = num[0] -> 只有一间房屋
dp[1] = max(num[0],num[1]) -> 只有2间房屋,选择金额最高的偷窃
最终答案即为dp[n-1]
<!--动态规划
时间复杂度是O(n),空间复杂度是O(n)
-->
class Solution {
public int rob(int[] nums) {
if (nums == null || nums.length == 0) {
return 0;
}
int length = nums.length;
if (length == 1) {
return nums[0];
}
int[] dp = new int[length];
dp[0] = nums[0];
dp[1] = Math.max(nums[0], nums[1]);
for (int i = 2; i < length; i++) {
dp[i] = Math.max(dp[i - 2] + nums[i], dp[i - 1]);
}
return dp[length - 1];
}
}
<!--空间优化
动态规划 + 滚动数组
时间复杂度是O(n),空间复杂度是O(1)
-->
class Solution {
public int rob(int[] nums) {
if (nums == null || nums.length == 0) {
return 0;
}
int length = nums.length;
if (length == 1) {
return nums[0];
}
int first = nums[0], second = Math.max(nums[0], nums[1]);
for (int i = 2; i < length; i++) {
int temp = second;
second = Math.max(first + nums[i], second);
first = temp;
}
return second;
}
}