Leetcode 213. 打家劫舍 II 动态规划

原题链接：Leetcode 213. 打家劫舍 II

class Solution {
public:
    int rob(vector<int>& nums) {
        int n = nums.size();
        if (n == 1)
            return nums[0];
        if (n == 2)
            return max(nums[0], nums[1]);
        // 如果偷了第一家，就不能偷最后一家
        int dp[n - 1];
        dp[0] = nums[0];
        dp[1] = nums[0];
        for (int i = 2; i < n - 1; i++) {
            dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);
        }
        // 如果从第二家开始偷，就没事
        int dp2[n];
        dp2[0] = 0;
        dp2[1] = nums[1];
        dp2[2] = max(nums[1], nums[2]);
        for (int i = 3; i < n; i++) {
            dp2[i] = max(dp2[i - 2] + nums[i], dp2[i - 1]);
        }
        return max(dp[n - 2], dp2[n - 1]);
    }
};