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

2020-02-19 16:39:53 浏览数 (1)

这道题目 https://cloud.tencent.com/developer/article/1581788 一样,改进了一点,就是首尾也是相邻的。

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

代码语言:javascript复制
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]));
        
    }
};

0 人点赞