213. House Robber II

213. House Robber II_第1张图片

解题思路:若第一座房子和最后一座房子只抢其一,去掉不抢的那栋,此题就回归到198 House Robber 的解!


代码:

class Solution {
        #define _MAX(a,b) ((a)>(b))?(a):(b)  
public:  
    int firstRob(vector<int>& nums,int n) {//此函数取自198 House Robber,仅增加了参数列表中的int n
        if(n==0)return 0;
        if(n==1)return nums[0];
        int d[n]={0};
        d[0]=nums[0];
        d[1]=_MAX(nums[0],nums[1]);
        for(int i=2;i<n;++i){
            d[i]=_MAX(d[i-2]+nums[i],d[i-1]);
        }
        return d[n-1];
    }
public:
    int rob(vector<int>& nums) {//动态规划,第一座房子和最后一座房子只能抢其一,然后删掉不抢的那栋,回到198 House Robber的解!
        int n=nums.size();
        if(n==0)return 0;
        if(n==1)return nums[0];
        int rob1=firstRob(nums,n-1);//不抢最后一座房子
        nums.erase(nums.begin());
        int rob2=firstRob(nums,n-1);//不抢第一座房子
        return _MAX(rob1,rob2);
    }
};


你可能感兴趣的:(LeetCode,C++)