Leetcode 740.删除与获得点数（Delete and Earn）

最新推荐文章于 2025-03-06 10:30:34 发布

原创最新推荐文章于 2025-03-06 10:30:34 发布 · 387 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#leetcode #算法

Leetcode 专栏收录该内容

595 篇文章

订阅专栏

本文详细解析了LeetCode 740题“删除与获得点数”的算法思路，通过将问题转化为打家劫舍问题，采用动态规划方法求解。介绍了两种解题策略，一种是直接使用数组重构，另一种是利用字典记录数值出现频率，再进行排序，以提高效率。

Leetcode 740.删除与获得点数

1 题目描述（Leetcode题目链接）

给定一个整数数组 nums ，你可以对它进行一些操作。
每次操作中，选择任意一个 nums[i] ，删除它并获得 nums[i] 的点数。之后，你必须删除每个等于 nums[i] - 1 或 nums[i] + 1 的元素。
开始你拥有 0 个点数。返回你能通过这些操作获得的最大点数。

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

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

注意:

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

2 题解

重构数组可以得到每个下标对应的点数之和，这就变成了打家劫舍问题。

class Solution:
    def deleteAndEarn(self, nums: List[int]) -> int:
        new_nums = [0] * 10001
        for num in nums:
            new_nums[num] += num
        DP = [0] * 10001
        DP[1] = new_nums[1]
        for i in range(2, len(new_nums)):
            DP[i] = max(DP[i-1], DP[i-2] + new_nums[i])
        return DP[-1]

但是效率比较低，用字典来记录每个数字的个数再排序。同样是转换成了打家劫舍问题，效率提升了不少。

class Solution:
    def deleteAndEarn(self, nums: List[int]) -> int:
        if not nums: return 0
        new_nums = list(dict(collections.Counter(nums)).items())
        new_nums.sort()
        length = len(new_nums)
        dp = [0]*(length + 1)
        dp[1] = new_nums[0][0]*new_nums[0][1]
        for i in range(2, length + 1):
            if new_nums[i-1][0] - 1 == new_nums[i-2][0]:
                dp[i] = max(dp[i-2]+new_nums[i-1][0]*new_nums[i-1][1], dp[i-1])
            else:
                dp[i] = max(dp[i-1]+new_nums[i-1][0]*new_nums[i-1][1], dp[i-1])    
        return dp[-1]