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];
    }
};
相关推荐
淡海水8 小时前
07-04-并发-ConcurrentBag-T-工作窃取WorkStealing算法
开发语言·算法·c#·bag·concurrent·workstealing
leobertlan9 小时前
好玩系列:训练一个神经网络模型指导小孩玩游戏2-大局观教练
算法
Tisfy10 小时前
LeetCode 3876.构造奇偶一致的数组 II:三种情况分类讨论(其实还是脑筋急转弯)
算法·leetcode·题解·脑筋急转弯
lvwangshu10 小时前
P7668 [JOI 2018 Final] 团子制作 / Dango Maker 题解
动态规划·题解·贪心·性质题·思维好题
乐迪信息10 小时前
智慧港口船舶AI算法实现在线状态监测
大数据·人工智能·深度学习·算法·计算机视觉
木井巳12 小时前
【BFS/DFS 解决 FloodFill 算法】太平洋大西洋水流问题
java·算法·leetcode·深度优先·广度优先·宽度优先·推荐算法
心抵鹊12 小时前
归并排序之翻转对(hard)
数据结构·算法
白山编程大哥12 小时前
Java 集合算法:从排序、查找到底层原理的实战指南
java·python·算法
shehuiyuelaiyuehao13 小时前
算法34,位运算符操作,总结
算法
Navigator_Z13 小时前
LeetCode //C - 1224. Maximum Equal Frequency
c语言·算法·leetcode