[leetcode] House Robber II

最新推荐文章于 2022-11-01 19:16:12 发布

cocolalala

最新推荐文章于 2022-11-01 19:16:12 发布

阅读量323

点赞数

分类专栏：算法文章标签：算法 c++ leetcode

本文链接：https://blog.csdn.net/yezizp/article/details/45936345

版权

算法专栏收录该内容

38 篇文章 0 订阅

订阅专栏

和I类似，II变成了环形，即首尾只能取一个。调用两次即可。代码如下：

int rob(vector<int>& nums) {
        if(nums.empty())
            return 0;
        if(nums.size() == 1)
            return nums[0];
        int val = *prev(nums.end());
        nums.pop_back();
        int result = robLine(nums);
        nums.erase(nums.begin());
        nums.push_back(val);
        return max(result, robLine(nums));
    }
    int robLine(vector<int> &num) {
        if(num.empty())
            return 0;
        int pre = 0, cur = num[0];
        int result = cur;
        for(int i = 1; i < num.size(); ++i){
            result = max(result, pre + num[i]);
            pre = cur;
            cur = result;
        }
        return result;
    }