[C++] LeetCode 213. 打家劫舍 II

题目

注意事项: 这是 打家劫舍 的延伸。
在上次盗窃完一条街道之后，窃贼又转到了一个新的地方，这样他就不会引起太多注意。这一次，这个地方的所有房屋都围成一圈。这意味着第一个房子是最后一个是紧挨着的。同时，这些房屋的安全系统与上次那条街道的安全系统保持一致。
给出一份代表每个房屋存放钱数的非负整数列表，确定你可以在不触动警报的情况下盗取的最高金额。

题解

由于是一个环形，所以需要考虑到第一个和最后一个，然后可以分两种情况来用动态规划。第一种为第一家偷的情况下，最大盗取金额，第二种为第一家不偷的情况下最大盗取金额。取二者最大值即为本题答案。

代码

class Solution {
public:
    int rob(vector<int>& nums) {
        int n=nums.size();
        if(n==0) return 0;
        vector<int> dp(n,0);
        int res=0;
        dp[0]=nums[0];
        for(int i=1;i<n;i++){
            if(i<2) dp[i]=max(dp[i-1],nums[i]);
            else dp[i]=max(dp[i-2]+nums[i],dp[i-1]);
        }
        res=n==1?nums[0]:dp[n-2];
        dp[0]=0;
        for(int i=1;i<n;i++){
            if(i<2) dp[i]=nums[i];
            else dp[i]=max(dp[i-2]+nums[i],dp[i-1]);
        }
        res=max(res,dp[n-1]);
        return res;
    }
};