LeetCode No.198 打家劫舍Ⅰ && LeetCode 面试17.16 按摩师问题
你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。
示例 :
输入: [1,2,3,1]
输出: 4
解释: 偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
偷窃到的最高金额 = 1 + 3 = 4 。
按摩师问题是一样的就不再复制题干了。
class Solution {
public:
int massage(vector<int>& nums) {
int n = nums.size();
if(n==0) return 0;
if(n==1) return nums[0];
int dp[n];
dp[0] = nums[0];
dp[1] = max(nums[0],nums[1]);
for(int i = 2;i<n;i++){
dp[i] = max(dp[i-1],dp[i-2]+nums[i]);
}
return dp[n-1];
}
};
时间复杂度O(n)。
空间复杂度O(n)。
LeetCode No.198 执行时间 4 ms
状态转移方程:dp[i] = max(dp[i-1],dp[i-2]+nums[i]);
偷还是不偷这是一个问题
偷:取dp[i-2]+nums[i]
不偷:取dp[i-1]
因为偷完之后要等待一个晚上才能继续偷,所以所有的夜晚都可以表示为第一晚、第三晚。
第一晚已经抢了一次,那么选择可以分为:
①第三晚抢nums[i]
并加入第一晚的所有金额之中dp[i-2]
②第三晚不打草惊蛇保持一如既往的金额dp[i-1]
优化:
上述代码用了n的空间,其实并不用那么大的空间就可以做出来:
class Solution {
public:
int massage(vector<int>& nums) {
int dp0 = 0;
int dp1 = 0;
for(int i =