1. 程式人生 > >213. House Robber II的C++解法(dp)

213. House Robber II的C++解法(dp)

題目描述: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;
    }
};