leetcode 213. House Robber II

和第198题没有本质区别。

定义dp数组,n是房间总个数。dpi表示从第0号房间一直到第i号房间(包含第i号房间)可以偷到的最大金额,按照这个定义,dpn-1就是答案。dp0一直到dpn-2的求法和第198题一模一样,唯独dpn-1需要特别考虑。

对于最后一个房间(即第n-1号房间),只有两种可能,偷或者不偷。

  1. 如果不偷第n-1号房间,则dpn-1 = dpn-1-1
  2. 如果偷第n-1号房间,则除了像第198题一样不能偷第n-2号房间外,还要求一定不能偷第0号房间。问题可以转化为,不考虑第0号房间,只考虑从第1号房间一直到第n-1-2号房间。因此另外定义一个dp2数组。dp2i表示从第1号房间一直到第i号房间(包含第i号房间)可以偷到的最大金额,dp20不定义。dpn-1 = dp2n-1-2 + numsn-1

dpn-1取前述两种情况的最大值,dpn-1 = max(dpn-1-1, dp2n-1-2 + numsn-1);

第一版代码

cpp 复制代码
class Solution {
public:
    int rob(vector<int>& nums) {
        int n = nums.size();
        //dp[i]表示从第0号房间一直到第i号房间(包含第i号房间)可以偷到的最大金额
        //按照这个定义,dp[n-1]就是答案。
        vector<int> dp(n);
        dp[0] = nums[0];
        if(n ==1)   return dp[0];
        dp[1] = max(nums[0],nums[1]);
        if(n ==2)   return dp[1];
        //dp[0]一直到dp[n-2]的求法和第198题一模一样,唯独dp[n-1]需要特别考虑。
        for(int i = 2;i < n -1;i++){
            dp[i] = max(dp[i-2]+nums[i],dp[i-1]);
        }

        //dp2[i]表示从第1号房间一直到第i号房间(包含第i号房间)可以偷到的最大金额,dp2[0]不定义
        vector<int> dp2(n);
        dp2[1] = nums[1];
        dp2[2] = max(nums[1],nums[2]);
        for(int i = 3; i < n;i++){
            dp2[i] = max(dp2[i-2]+nums[i],dp2[i-1]);
        }

        dp[n-1] = max(dp2[n-1-2]+nums[n-1],dp[n-1-1]);
        return dp[n-1];
    }
};

求dp2和求dp可以在一次遍历中求取。

cpp 复制代码
class Solution {
public:
    int rob(vector<int>& nums) {
        int n = nums.size();
        //dp[i]表示从第0号房间一直到第i号房间(包含第i号房间)可以偷到的最大金额
        //按照这个定义,dp[n-1]就是答案。
        vector<int> dp(n);
        dp[0] = nums[0];
        if(n ==1)   return dp[0];
        dp[1] = max(nums[0],nums[1]);
        if(n ==2)   return dp[1];
        if(n ==3)   return max(max(nums[0],nums[1]),nums[2]);

        //dp2[i]表示从第1号房间一直到第i号房间(包含第i号房间)可以偷到的最大金额,dp2[0]不定义
        vector<int> dp2(n);
        dp2[1] = nums[1];
        dp2[2] = max(nums[1],nums[2]);
        for(int i = 2;i < n -1;i++){
            if(i>=3)
                dp2[i] = max(dp2[i-2]+nums[i],dp2[i-1]);
            dp[i] = max(dp[i-2]+nums[i],dp[i-1]);
        }
        //dp[n-1]需要特别考虑
        dp[n-1] = max(dp2[n-1-2]+nums[n-1],dp[n-1-1]);
        return dp[n-1];
    }
};
相关推荐
Jerry12 分钟前
LeetCode 203. 移除链表元素
算法
地平线开发者25 分钟前
征程 6 | 工具链 QAT ObserverBase 源码解析
算法
地平线开发者1 小时前
【地平线 征程 6 工具链进阶教程】QAT 训练常见问题和排查
算法
地平线开发者1 小时前
征程 6 | 直方图量化配置与校准实例
算法
地平线开发者1 小时前
征程 6E/M Matrix 开发评板使用系列(一):开箱与点亮
算法·自动驾驶
Jerry2 小时前
LeetCode 59. 螺旋矩阵 II
算法
tachibana22 小时前
hot100 回文链表(234)
java·网络·数据结构·leetcode·链表
可编程芯片开发2 小时前
基于FOC控制器的BLDC无刷直流电机控制系统matlab编程与仿真
算法
aaaameliaaa2 小时前
进制练习题【找出只出现一次的数字、交换两个变量(不创建临时变量)、统计二进制中1的个数、打印整数二进制的奇数位和偶数位、求两个数二进制中不同位的个数】
c语言·数据结构·笔记·算法