你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。
思路 动态规划
两个数组,nums[n] 存储房屋现金,dp[n] 存储前n间房屋最佳抢劫方案
第n间房屋抢劫方案分为,抢劫第n间房屋,以及不抢劫第n间房屋(即前n-1间房屋抢劫最佳方案)
得到递推公式:dp[i]=max(dp[i−2]+nums[i],dp[i−1])。
由于 dp[i] 只取决于 dp[i−1] 和 dp[i−2],和更早的状态无关,因此可以使用滚动数组的思想,只保留最近的两项,将空间复杂度降到 O(1)。
public class Solution {
public int Rob(int[] nums) {
int n = nums.Length;
if(n == 1)
return nums[0];
int[] dp = new int[n];
dp[0] = nums[0];
dp[1] = Math.Max(nums[0], nums[1]);
for(int i = 2; i < n; i++)
dp[i] = Math.Max(dp[i - 1], dp[i - 2] + nums[i]);
return dp[n - 1];
}
}
滚动数组解法
public class Solution {
public int Rob(int[] nums) {
int n = nums.Length;
if (n == 1) {
return nums[0];
}
int prev = nums[0], curr = Math.Max(nums[0], nums[1]);
for (int i = 2; i < n; i++) {
int next = Math.Max(prev + nums[i], curr);
prev = curr;
curr = next;
}
return curr;
}
}
复杂度分析
-
时间复杂度:O(n),其中 n 是数组 nums 的长度。状态数是 O(n),每个状态的计算时间是 O(1),因此时间复杂度是 O(n)。
-
空间复杂度:O(n) 或 O(1),其中 n 是数组 nums 的长度。空间复杂度取决于实现方式,不优化空间的实现需要创建长度为 n 的数组 dp 因此空间复杂度是 O(n),优化空间的实现的空间复杂度是 O(1)。