213. House Robber II

最新推荐文章于 2021-09-21 10:16:04 发布

密函一封

最新推荐文章于 2021-09-21 10:16:04 发布

阅读量587

点赞数

分类专栏： leetcode C/C++ 文章标签： leetcode C++

本文链接：https://blog.csdn.net/jingmiaa/article/details/51320839

版权

C/C++ 同时被 2 个专栏收录

68 篇文章 1 订阅

订阅专栏

leetcode

32 篇文章 0 订阅

订阅专栏

解题思路：若第一座房子和最后一座房子只抢其一，去掉不抢的那栋，此题就回归到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);
    }
};