leetcode: 213. 打家劫舍 II

题目描述

你是一个专业的小偷，计划偷窃沿街的房屋，每间房内都藏有一定的现金。这个地方所有的房屋都 围成一圈 ，这意味着第一个房屋和最后一个房屋是紧挨着的。同时，相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警 。
给定一个代表每个房屋存放金额的非负整数数组，计算你 在不触动警报装置的情况下 ，能够偷窃到的最高金额。

示例 1：

输入：nums = [2,3,2]
输出：3
解释：你不能先偷窃 1 号房屋（金额 = 2），然后偷窃 3 号房屋（金额 = 2）, 因为他们是相邻的。

示例 2：

输入：nums = [1,2,3,1]
输出：4
解释：你可以先偷窃 1 号房屋（金额 = 1），然后偷窃 3 号房屋（金额 = 3）。
     偷窃到的最高金额 = 1 + 3 = 4 。

示例 3：

输入：nums = [0]
输出：0

提示：

• 1 <= nums.length <= 100
• 0 <= nums[i] <= 1000

---

状态转移方程

• 题解利用动态规划，难点在于得出动态转移方程
• 假设偷窃房屋的下标范围是 [start, end]，用dp[i] 表示在下标范围 [start, i] 内可以偷窃到的最高总金额，那么就有如下的状态转移方程：
  $$dp[i]=max(dp[i-2]+nums[i],dp[i-1])$$

边界条件为：
$$\{\begin{array}{ll}dp[start]=nums[start]& \text{if }只有一间房屋，则偷窃该房屋\\ dp[start+1]=max(nums[start],nums[start+1])& \text{if }只有两间房屋，偷窃其中金额较高的房屋\end{array}$$

• 计算得到dp[end] 即为下标范围[start,end] 内可以偷窃到的最高总金额。

• 分别取 (start, end) = (0, n − 2) 和 (start, end) = (1, n − 1) 进行计算，取两个 dp[end] 中的最大值，即可得到最终结果。

---

结果

class Solution {
public:
    int rebRes(vector<int>& nums, int begin, int end)
    {   //利用回滚数组
        int res = nums[begin], res2 = max(res, nums[begin+1]);
        for (int i = begin + 2; i <= end; ++i)
        {
            int temp = res2;
            res2 = max(res2, res + nums[i]);
            res = temp;
        }
        return res2;
    }

    int rob(vector<int>& nums) {
        int len = nums.size();
        if (0 == len) return 0;
        if (1 == len) return nums[0];
        if (2 == len) return max(nums[0], nums[1]);
        return max(rebRes(nums, 0, len-2), rebRes(nums, 1, len-1));
    }
};