打家劫舍 II

在这里插入图片描述
相较于打家劫舍,算是一个升级版。既然是打家劫舍的升级版,那我们就先找与原版之间的联系。做过打家劫舍之后我们知道状态转移方程是dp[i]=max(dp[i-1],dp[i-2]+nums[i-1])。
然后我们再来看这道题,由相邻的一家变成了相邻的两家。那么就相当于把原来的一条直线首尾相连变成了一个环,这句话你细品,哈哈。所以我们要做的就是把这个环怎么变成直链,因为首尾相连,所以我们如果偷第一家就肯定不能偷最后一家,也就是说我们分成两种情况:**我们今晚就在第一家到第n-1家里面偷,和我们今晚就在第2家到第n家里面偷。**然后把这两种情况分别求出来的最大值再次比较,看看那种能偷到的利益更大,代码如下:

class Solution {
public:
    int rob(vector<int>& nums) {
        if(nums.size()==1)
            return nums[0];
        int n=nums.size();
        vector<int>a(n+1),b(n+1);
        for(int i=2;i<n+1;i++){
            a[i]=max(a[i-1],a[i-2]+nums[i-2]);
            b[i]=max(b[i-1],b[i-2]+nums[i-1]);
        }
        return max(a[n],b[n]);
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章