面试经典 150 题 2.1 —（一维动态规划）— 213. 打家劫舍 II

最新推荐文章于 2024-05-04 17:31:20 发布

BreezeChasingDrizzle

最新推荐文章于 2024-05-04 17:31:20 发布

阅读量118

点赞数 3

分类专栏： leetcode 文章标签：动态规划

本文链接：https://blog.csdn.net/weixin_44032178/article/details/136407542

版权

leetcode 专栏收录该内容

174 篇文章 0 订阅

订阅专栏

213. 打家劫舍 II

在这里插入图片描述

class Solution {
public:
    /*
    环状排列意味着第一个房子和最后一个房子中只能选择一个偷窃，
    因此可以把此环状排列房间问题约化为两个单排排列房间子问题(198)：
    在不偷窃第一个房子的情况下（即 nums[1:]），最大金额是p1；
    在不偷窃最后一个房子的情况下（即 nums[:n-1]），最大金额是p2。
    综合偷窃最大金额： 为以上两种情况的较大值，即 max(p1,p2)。
    */
    int rodUnidirection(vector<int> nums, int startIndex, int endIndex){
        vector<int> dp(endIndex-startIndex+2, 0);
        dp[startIndex] = nums[startIndex];
        for(int dpIndex = startIndex + 1; dpIndex <= endIndex; dpIndex++){
            if(dpIndex == 1){
                dp[1] = max(nums[startIndex], nums[startIndex + 1]);
            }
            else{
                dp[dpIndex] = max((dp[dpIndex - 2] + nums[dpIndex]), dp[dpIndex - 1]);
            }
        }
        return dp[endIndex];
    }
    int rob(vector<int>& nums) {
        int numsSize = nums.size();
        if(numsSize == 0){
            return 0;
        }
        if(numsSize == 1){
            return nums[0];
        }
        int max0ToN_2 = rodUnidirection(nums, 0, numsSize - 2);
        int max1ToN_1 = rodUnidirection(nums, 1, numsSize - 1);
        return max(max0ToN_2, max1ToN_1);
    }
};

class Solution {
public:
    /*
    环状排列意味着第一个房子和最后一个房子中只能选择一个偷窃，
    因此可以把此环状排列房间问题约化为两个单排排列房间子问题(198)：
    在不偷窃第一个房子的情况下（即 nums[1:]），最大金额是p1；
    在不偷窃最后一个房子的情况下（即 nums[:n-1]），最大金额是p2。
    综合偷窃最大金额： 为以上两种情况的较大值，即 max(p1,p2)。
    */
    int rodUnidirection(vector<int> nums, int startIndex, int endIndex){
        vector<int> dp(endIndex-startIndex+2, 0);
        dp[startIndex] = nums[startIndex];
        // dp中的坐标不管是从0到numsSize-2还是从1到numsSize-1，不需要非得从第一个位置开始存储，剩下的那个位置不用管，最后一个都是dp[endIndex]
        for(int dpIndex = startIndex + 1; dpIndex <= endIndex; dpIndex++){
            if(dpIndex == 1){
                dp[1] = max(nums[startIndex], nums[startIndex + 1]);
            }
            else{
                dp[dpIndex] = max((dp[dpIndex - 2] + nums[dpIndex]), dp[dpIndex - 1]);
            }
        }
        return dp[endIndex];
    }
    int rob(vector<int>& nums) {
        int numsSize = nums.size();
        if(numsSize == 0){
            return 0;
        }
        if(numsSize == 1){
            return nums[0];
        }
        int max0ToN_2 = rodUnidirection(nums, 0, numsSize - 2);
        int max1ToN_1 = rodUnidirection(nums, 1, numsSize - 1);
        return max(max0ToN_2, max1ToN_1);
    }
};