leetcode213. 打家劫舍 II

一:题目

二:上码

class Solution {
public:

    /**
        思路:
            1.既然成环了,我们如果选取得一条偷取路径是从头开始得那么我们就不能偷取最后一个,那就不算最后一个偷取一遍
            2.同理我们也可以不算第一个 偷取一遍计算一次偷取得结果
    */
    
    int rob(vector<int>& nums) {
        if(nums.size() == 0) return 0;
        if(nums.size() == 1) return nums[0];

        int ans1 = steal(nums,0,nums.size()-2);//不算最后一个 这里传递是下标
        int ans2 = steal(nums,1,nums.size()-1);//不算第一个

        return ans1 > ans2 ? ans1 : ans2;
    }

    int steal(vector<int> & nums,int start,int end) {
         if (end == start) return nums[start];
        vector<int> dp(nums.size(),0);

        dp[start] = nums[start];
        dp[start+1] = max(nums[start],nums[start+1]);

        for(int i = start+2; i <= end; i++) {
            dp[i] = max(dp[i-1],dp[i-2]+nums[i]);
        }
        
        return dp[end];
    }
};