LeetCode213. 打家劫舍 II

标签：lin   int end   ima   blog   src   htm   方案   int   最大   

这题是LeetCode198. 打家劫舍的进阶版，除了要求不能
打劫相邻的房子以外，还不能同时打劫第一个和最后一个房子。因为所有房子的排列是环形的，所以实际上第一个房子和
最后一个房子也是相邻的。

我们也用dp[i]表示打劫到第i个房子能获得的最大收益，要打劫到最后一个房子，就得不打劫第一个房子，所以我们需要考虑两种
情况，打劫第一个房子和不打劫第一个房子。

如果我们打劫了第一个房子，那么就不能考虑最后一个房子了，可打劫的房子从第一个房子到倒数第二个房子。

如果我们不打劫第一个房子，那么我们就可以打劫最后一个房子，可打劫的房子从第二个房子到最后一个房子。

实际上，对于上述两种情况，由于有了打劫的房子的范围，我们只需要对于打劫的范围内的房子用第198题的方式打劫就好了，因为我们已经
分情况（划分打劫区间）考虑了首尾相邻的问题，所以对于这两种情况我们分别求出在对应区间内打劫到的最大值，然后对于这两个结果
再取一个最大值就是最终的结果了。

代码如下：

class Solution {
public:
    int robHouses(vector<int>& nums, int start, int end) {                  //打劫区间为[start, end]（注意左右都是闭）时可打劫到的最大收益，这部分和198题一样
        vector<int> dp(nums.size());
        dp[start] = nums[start], dp[start + 1] = max(nums[start], nums[start + 1]);      //确定边界情况
        for(int i = start + 2; i <= end; ++i) {
            dp[i] = max(dp[i - 1], dp[i - 2] + nums[i]);                  //打劫到第i个房子时的最大值为（不打劫第i个房子，打劫第i个房子）两种打劫方案中的较大值
        }
        return dp[end];                                          //dp[end]就是[start, end]区间能打劫到的最大值
    }

    int rob(vector<int>& nums) {
        int n = nums.size();
        if(n == 0) {                                    //几个特殊情况先特判一下
            return 0;
        }
        if(n == 1) {
            return nums[0];
        }
        if(n == 2) {
            return max(nums[0], nums[1]);
        }
        return max(robHouses(nums, 0, n - 2), robHouses(nums, 1, n - 1));          //（打劫第一个房子，不打劫第一个房子）两个方案中的较大值就是最终能打劫到的最大值
    }
};

LeetCode213. 打家劫舍 II

标签：lin   int end   ima   blog   src   htm   方案   int   最大   

原文地址：https://www.cnblogs.com/linrj/p/13504375.html