leetcode--213. 打家劫舍 II

213. 打家劫舍 II
     你是一个专业的小偷，计划偷窃沿街的房屋，每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈，这意味着第一个房屋和最后一个房屋是紧挨着的。同时，相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。

给定一个代表每个房屋存放金额的非负整数数组，计算你在不触动警报装置的情况下，能够偷窃到的最高金额。

示例 1:

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

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

思路：做两次198题的动态规划，一次是[0,n-2]，一次是[1,n-1]，取二者较大

class Solution {
public:
    int rob(vector<int>& nums) {
        int n = nums.size();
        if (1 == n) return nums[0];
        int uncheck = 0;
        int checked = 0;
        int ans1 = 0;
        for (int i=0; i<n-1; i++) {
            ans1 = max(checked, uncheck+nums[i]);
            uncheck = checked;
            checked = ans1;
        }
        uncheck = 0;
        checked = 0;
        int ans2 = 0;
        for (int i=1; i<n; i++) {
            ans2 = max(checked, uncheck+nums[i]);
            uncheck = checked;
            checked = ans2;
        }
        return max(ans1, ans2);
    }
};
/*4ms,7.9MB*/

优化版：

class Solution {
public:
    int rob(vector<int>& nums) {
        int n = nums.size();
        if (1 == n) return nums[0];
        return max(robRange(nums, 0, n-2), robRange(nums, 1, n-1));
    }
    int robRange(vector<int>& nums, int start, int end) {
        int uncheck = 0;
        int checked = 0;
        for (int i=start; i<=end; i++) {
            int ans = max(checked, uncheck+nums[i]);
            uncheck = checked;
            checked = ans;
        }
        return checked;
    }
};
/*0ms,7.9MB*/