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];
    }
};
相关推荐
不忘不弃7 分钟前
计算pi的近似值
算法
码云骑士7 分钟前
12-GIL不是性能杀手(下)-绕过GIL的三种方案与决策树
算法·决策树·机器学习
一只齐刘海的猫10 分钟前
【Leetcode】无重复字符的最长子串
算法·leetcode·职场和发展
行智科技12 分钟前
FAST-LIVO2 源码精读(二):环境搭建与编译避坑
算法·ubuntu·自动驾驶·slam
插件开发22 分钟前
vs2015 cuda c++ cdpSimplePrint范例,递归功能实现演示
linux·c++·算法
Tisfy23 分钟前
LeetCode 2130.链表最大孪生和:转数组 / 快慢指针+链表翻转(O(1))
算法·leetcode·链表·题解
来自于狂人1 小时前
第5章 记忆管理——让Agent记住事情
人工智能·算法·语言模型·自然语言处理
CHHH_HHH1 小时前
【C++】哈希表原理与实战:从冲突解决到性能优化
开发语言·数据结构·c++·学习·算法·哈希算法·散列表
sali-tec1 小时前
C# 基于OpenCv的视觉工作流-章84-包胶有无检测
图像处理·人工智能·opencv·算法·计算机视觉
Irissgwe1 小时前
数据结构-排序
数据结构·算法·排序算法