力扣740. 删除并获得点数(动态规划)

文章讲述了如何使用动态规划方法解决一个关于整数数组的删除操作问题,目标是通过删除元素并获得点数,同时避免连续元素。通过计算非相邻元素的最大点数,转化为背包问题中的状态转移方程。
摘要由CSDN通过智能技术生成

题目描述:

给你一个整数数组 nums ,你可以对它进行一些操作。

每次操作中,选择任意一个 nums[i] ,删除它并获得 nums[i] 的点数。之后,你必须删除 所有 等于 nums[i] - 1 和 nums[i] + 1 的元素。

开始你拥有 0 个点数。返回你能通过这些操作获得的最大点数。

示例 1:

输入:nums = [3,4,2]
输出:6
解释:
删除 4 获得 4 个点数,因此 3 也被删除。
之后,删除 2 获得 2 个点数。总共获得 6 个点数。

示例 2:

输入:nums = [2,2,3,3,3,4]
输出:9
解释:
删除 3 获得 3 个点数,接着要删除两个 2 和 4 。
之后,再次删除 3 获得 3 个点数,再次删除 3 获得 3 个点数。
总共获得 9 个点数。

提示:

  • 1 <= nums.length <= 2 * 104
  • 1 <= nums[i] <= 104

思路:

根据题目意思我们可以很容易得到,如果我们要选择删除x,那么我们会将所有的x+1和x-1的元素全部删掉并获得x点数,换句话说,我们选择不能同时同时获得x和x+1点数或者是x和x-1点数。因为,如果获得了x点数,x+1的元素与x-1的元素就已经全部不存在了,而如果有多个x,我们就可以获得这些x总和的点数。

我们可以用sum[x]表示元素x的总和点数。

那么这个题和打家劫舍就一样了,求不能选相邻元素的最大点数

转化成为背包的思想--选或者不选

选x,就要加上所有x总和的点数,这个状态是由x-2转移过来的

不选x,那么上一个状态就是x-1

用f[x]表示在0--x里面选不相邻的值,获得的最大点数。

f[x]=max(f[x-1],f[x-2]+sum[x])

代码:

class Solution {
public:
    int deleteAndEarn(vector<int>& nums) {
        int len=nums.size();
        int max_v=0;//记录nums数组里最大的数,也就是数组sum的下标范围
        for(int i=0;i<len;i++){
            if(nums[i]>max_v){
                max_v=nums[i];
            }
        }
        vector<int> sum(max_v+10,0);
        for(int i=0;i<len;i++){
            sum[nums[i]]+=nums[i];//sum[nums[i]]表示值为nums[i]的总和
        }
        vector<int> f(max_v+1,0);
        
        f[0]=sum[0];
        f[1]=max(sum[1],sum[0]);
        for(int i=2;i<=max_v;i++){
            f[i]=max(f[i-1],f[i-2]+sum[i]);
        }
        return f[max_v];
    }
};

评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值