打家劫舍 (dp)

23 篇文章 0 订阅

下定决心,好好过一天 ~
题目传送
在这里插入图片描述
这是简单版本。
思路:
如果拿第 i 个物品,一定可以拿第 i-2 或者 i-3 个物品不然则亏损(没必要),那么动态方程就出来了,直接i从0往后遍历即可。dp[i] = max(nums[i] + dp[i-2] , nums[i] + dp[i-3]),输出dp数组中最大值即可

class Solution {
public:
    int rob(vector<int>& nums) {
        int dp[1005] = {0};
        int len = nums.size();
        dp[0] = nums[0];
        if(len > 1) dp[1] = nums[1];
        if(len > 2) dp[2] = dp[0] + nums[2];
        int Max = max({dp[0],dp[1],dp[2]});
        for(int i = 3;i < len;i++)
        {
            int l = i-2,r = i-3;
            dp[i] = max(nums[i]+dp[l],nums[i]+dp[r]);
            Max = max(Max,dp[i]);
        }
        // cout << Max << endl;
        return Max;
    }
};

升级版:
在这里插入图片描述
和上面思路差不多,只是,当前是一个环。那么第一个和最后一个不能同时选。
那么就分俩种情况dp,取最大值即可,就是dp从2到n,和从1到n-1。这样肯定满足条件

class Solution {
public:
    int rob(vector<int>& nums) {
        int dp[1005] = {0};
        int len = nums.size();
        dp[0] = nums[0];
        if(len > 1) dp[1] = nums[1];
        if(len > 2 && len != 3) dp[2] = dp[0] + nums[2];
        int Max = max({dp[0],dp[1],dp[2]});
        for(int i = 3;i < len-1;i++)
        {
            int l = i-2,r = i-3;
            dp[i] = max(nums[i]+dp[l],nums[i]+dp[r]);
            Max = max(Max,dp[i]);
        }
        fill(dp,dp+1005,0);
        dp[0] = 0;
        if(len > 1) dp[1] = nums[1];
        if(len > 2) dp[2] = nums[2];
        Max = max({Max,dp[1],dp[2]});
        for(int i = 3;i < len;i++)
        {
            int l = i-2,r = i-3;
            dp[i] = max(nums[i]+dp[l],nums[i]+dp[r]);
            Max = max(Max,dp[i]);
        }
        // cout << Max << endl;
        return Max;
    }
};
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值