Leetcode-213 打家劫舍 II

本题链接

相关笔记
Leetcode-196 打家劫舍
Leetcode-337 打家劫舍Ⅲ

本题思路

本题关键在于第一家（0）和最后一家（nums.size()-1）不能同时偷
在这个前提下，我们把这个问题分解为两个子问题0~nums.size-2和1~nums.size()-1对于偷这两个范围，我们可以获得的最大值。

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]);
        vector<int> dp_a(nums.size(),0);
        vector<int> dp_b(nums.size(),0);
        dp_a[0]=nums[0];
        dp_a[1]=max(nums[0],nums[1]);        
        dp_b[0]=nums[1];
        dp_b[1]=max(nums[1],nums[2]);
        for(int i=2;i<nums.size()-1;i++)
        {
            dp_a[i]=max(dp_a[i-1],dp_a[i-2]+nums[i]);
            dp_b[i]=max(dp_b[i-1],dp_b[i-2]+nums[i+1]);
        }
        return max(dp_a[nums.size()-2],dp_b[nums.size()-2]);
    }
};