力扣 213. 打家劫舍 II

最新推荐文章于 2024-07-21 08:37:44 发布

江湖人称贺行风

最新推荐文章于 2024-07-21 08:37:44 发布

阅读量323

点赞数

分类专栏：开始C++吧文章标签： leetcode 算法 c++ 动态规划

本文链接：https://blog.csdn.net/u010950122/article/details/132315640

版权

开始C++吧专栏收录该内容

180 篇文章 1 订阅

订阅专栏

题目来源：https://leetcode.cn/problems/house-robber-ii/description/

C++题解（来源代码随想录）：对于一个数组，成环的话主要有如下三种情况：（1）考虑不包含首尾元素；（2）考虑包含首元素，不包含尾元素；（3）考虑包含尾元素，不包含首元素。“考虑”包含尾元素，但不一定要选尾部元素！而情况二和情况三都包含了情况一了，所以只考虑情况二和情况三就可以了。

class Solution {
public:
    int rob(vector<int>& nums) {
        int len = nums.size();
        if(len == 1) return nums[0];
        if(len == 2) return max(nums[0], nums[1]);
        // 不偷最后一个房间
        vector<int> dp1(len-1);
        dp1[0] = nums[0];
        dp1[1] = max(nums[0], nums[1]);
        
        // 不偷第一个房间
        vector<int> dp2(len-1);
        dp2[0] = nums[1];
        dp2[1] = max(nums[1], nums[2]);

        for(int i = 2; i < len-1; i++) {
            dp1[i] = max(dp1[i-1], dp1[i-2]+nums[i]);
            dp2[i] = max(dp2[i-1], dp2[i-2]+nums[i+1]);
        }
        
        return max(dp1[len-2], dp2[len-2]);
    }
};

class Solution {
public:
    int rob(vector<int>& nums) {
        int len = nums.size();
        if(len == 1) return nums[0];
        if(len == 2) return max(nums[0], nums[1]);
        // 不偷最后一个房间
        vector<int> dp1(len-1);
        dp1[0] = nums[0];
        dp1[1] = max(nums[0], nums[1]);
        
        // 不偷第一个房间
        vector<int> dp2(len-1);
        dp2[0] = nums[1];
        dp2[1] = max(nums[1], nums[2]);

        for(int i = 2; i < len-1; i++) {
            dp1[i] = max(dp1[i-1], dp1[i-2]+nums[i]);
            dp2[i] = max(dp2[i-1], dp2[i-2]+nums[i+1]);
        }
        
        return max(dp1[len-2], dp2[len-2]);
    }
};