LeetCode 213. House Robber II (动态规划)

Posted 2021-03-09 Shendu.cc

题目

和这道题目 https://www.cnblogs.com/dacc123/p/12295924.html 一样，改进了一点，就是首尾也是相邻的。

那么我们在DP的时候，还要考虑第一个房子有没有被抢劫的情况。所以状态数组变成了DP[i][j][k]，j表示i个房子是否抢劫，k表示第1个房子是否抢劫。

class Solution {
public:
    int dp[100005][2][2];
    int rob(vector<int>& nums) {
        
        if(nums.size()==0)
            return 0;
        if(nums.size()==1)
            return nums[0];
        dp[0][0][0]=0;
        dp[0][1][1]=nums[0];
        
        dp[0][0][1]=-1;
        dp[0][1][0]=-1;
        
        for(int i=1;i<nums.size();i++)
        {
            if(dp[i-1][0][1]==-1)
                dp[i][1][1]=-1;
            else
                dp[i][1][1] = dp[i-1][0][1] + nums[i];
            
            dp[i][1][0] = dp[i-1][0][0] + nums[i];
            
            dp[i][0][1] = max(dp[i-1][1][1],dp[i-1][0][1]);
            
            dp[i][0][0] = max(dp[i-1][1][0],dp[i-1][0][0]);
        }
        
        return max(dp[nums.size()-1][1][0],max(dp[nums.size()-1][0][1],dp[nums.size()-1][0][0]));
        
    }
};