213. House Robber II的C++解法(dp)
阿新 • • 發佈:2019-01-02
題目描述:https://leetcode.com/problems/house-robber-ii/
在House Robber的基礎上進行考慮,如果打劫第0間,就不能打劫第n間,相當於求解[0,n-1]的最大值;如果打劫第n間,就只能從第1間開始打劫,相當於求解[1,n]的最大值,最後比較這兩個值哪個大即可。
class Solution { public: int rob(vector<int>& nums) { if (nums.size()==0) return 0; if (nums.size()==1) return nums[0]; if (nums.size()==2) return max(nums[0],nums[1]); return max(robber(nums,0,nums.size()-2),robber(nums,1,nums.size()-1)); } int robber(vector<int>& nums,int i,int j){ int a=nums[i]; int b=max(nums[i],nums[i+1]); for (int k=i+2;k<=j;k++) { int tmp=max(a+nums[k],b); a=b; b=tmp; } return b; } };