题解 | 打家劫舍(二)

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param nums int整型vector 
     * @return int整型
     */
    //分两种情况偷第一家还是不偷第一家
    int rob(vector<int>& nums) {
        // write code here
         int n=nums.size();
         if(n==1)
         {
            return nums[0];
         }
        vector<int> dp(n+1,0);
        //偷第一家
        dp[1]=nums[0];
        for(int i=2;i<n;i++)
        {
            dp[i]=max(dp[i-2]+nums[i-1],dp[i-1]);
        }
        int maxres=dp[n-1];
        //不偷第一家
        dp.assign(dp.size(), 0);
        dp[2]=nums[1];
        for(int i=3;i<=n;i++)
        {
            dp[i]=max(dp[i-2]+nums[i-1],dp[i-1]);
        }
    
        return max(dp[n],maxres);
    }
};