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