LeetCode第 213 题:打家劫舍II(C++)

213. 打家劫舍 II - 力扣(LeetCode)

LeetCode第 198 题:打家劫舍(C++)_qq_32523711的博客-CSDN博客的扩展。

数组首尾是一个圈,所以首尾也是相邻的。第一个想法是先别管是不是圈,反正就是第一个和最后一个不能同时抢

那我们把序列分为0 ~ n-1,1 ~ n两部分,分别计算,最后返回最大值就可以了。

class Solution {
public:
    int rob(vector& nums) {
        if(nums.size() == 0)    return 0;
        if(nums.size() == 1)    return nums[0];
        int pre = 0, cur = 0;
        for(int i = 0; i < nums.size()-1; ++i){//0~n-1
            int tmp = cur;
            cur = max(cur, pre+nums[i]);
            pre = tmp;
        }
        int res1 = cur;

        pre = 0, cur = 0;
        for(int i = 1; i < nums.size(); ++i){//1~n
            int tmp = cur;
            cur = max(cur, pre+nums[i]);
            pre = tmp;
        }
        return max(cur, res1);
    }
};

你可能感兴趣的:(leetcode,leetcode)