LeetCode213. House Robber II

题目链接:

https://leetcode.com/problems/house-robber-ii/

题目描述:

这道题就是在上一题(198. House Robber)的基础上加了一个条件,变成了环,所以如果抢了第一家,就不能抢最后一家。

思路:

我们可以分别求从第一家到倒数第二家的不相邻数据组合的最大值,从第二家到最后一家的不相邻数据组合的最大值。比较找出最终最大值。

代码:

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

你可能感兴趣的:(LeetCode,动态规划)