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]);
    }
};