代码随想录算法训练营第三十九天-动态规划-213. 打家劫舍 II

最新推荐文章于 2025-03-15 10:56:02 发布

taoyong001

最新推荐文章于 2025-03-15 10:56:02 发布

阅读量463

点赞数 6

文章标签：算法动态规划 c++ leetcode

本文链接：https://blog.csdn.net/taoyong001/article/details/145401015

版权

与上一题基本一样，只不过房间形成一个环，就需要在首尾考虑状况多一些
这不是多一些状况的问题，是完全不知道如何选择的问题
这种状况详细分析一下就是要分成三种情况
- 第一种：不考虑首元素，也不考虑尾元素，只考虑中间的元素，那么中间的元素只需要按照上一题的方式来考虑就可以了，因为剔除了首尾元素，肯定就不会出现连续抢劫店家的情况
- 第二种：不考虑尾元素，只考虑从到倒数第二家的情况
- 第三种：不考虑首元素，只考虑从第二家到最后一家的情况
- 这三种情况当中，第二种与第三种情况是包含了第一种情况的
- 只考虑第二种与第三种情况下，就会把一个环形问题转化成线性问题（同上一题一样了）
- 所以这个问题变变成去除尾元素的最值，与去除头元素最值的这两个值谁更大了

class Solution {
private:
    int commonRob(std::vector<int>& values) {
        if (values.size() == 1) {
            return values.at(0);
        }
        int dp[101];
        memset(dp, 0, sizeof(dp));
        dp[0] = values[0];
        dp[1] = std::max(values[0], values[1]);
        for (int i = 2; i < values.size(); ++i) {
            dp[i] = std::max(dp[i - 2] + values[i], dp[i - 1]);
        }
        return dp[values.size() - 1];
    }
public:
    int rob(vector<int>& nums) {
        if (nums.size() == 0) {
            return 0;
        }
        if (nums.size() == 1) {
            return nums.at(0);
        }
        std::vector<int> v1(nums.begin(), nums.end() - 1);
        std::vector<int> v2(nums.begin() + 1, nums.end());
        return std::max(commonRob(v1), commonRob(v2));
    }
};

汇总