(LeetCode 740)删除与获得点数 [DP + 打家劫舍变形]

740. 删除与获得点数
给定一个整数数组 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 个点数。
注意:

nums的长度最大为20000。
每个整数nums[i]的大小都在[1, 10000]范围内。


分析:
每个整数nums[i]的大小都在[1, 10000]范围内。
由此可知,我们可以使用一个int v[10010]大小的数组,记录每个数字nums[i]的和(nums[i] * 出现的次数)。
使用minnum 记录最小的nums[i]值,maxnum 记录最大的nums[i]值
因此问题就转换成了“打家劫舍”的题目了:
在编号问为minnum ~ maxnum 之间的房间里,每个房间里有金币数为v[i],你不能偷取连续房间的金币,问你最多可以偷取多少金币。

设dp[i]为偷取前i个房间可以获得的最大金币数,则:
dp[i] = max(dp[i-1], dp[i-2] + v[i])

在代码中,我们将v[] 和 dp[] 和并为一个数组,节约空间。

AC代码:

class Solution {
public:
    int deleteAndEarn(vector<int>& nums) {
        int n = nums.size();
        if(n == 0) return 0;
        if(n == 1) return nums[0];
        int dp[10010];
        //dp[] = v[]
        memset(dp,0,sizeof(dp));
        int maxnum = 0;
        int minnum = 10010;
        for(int i=0;i<n;i++)
        {
            maxnum = max(maxnum, nums[i]);
            minnum = min(minnum, nums[i]);
            dp[nums[i]] += nums[i];
        }
        //dp[] = dp[]
        for(int i=minnum;i<=maxnum;i++)
        {
            dp[i] = max(dp[i-1], ((i-2 > 0)? dp[i-2]:0)+dp[i]);
        }
        return dp[maxnum];
    }
};
  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值