力扣 213. 打家劫舍 II

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

力扣 213. 打家劫舍 II_第1张图片

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

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

你可能感兴趣的:(开始C++吧,leetcode,算法,c++,动态规划)