【Leetcode】740. Delete and Earn

最新推荐文章于 2021-10-23 05:24:25 发布

Vincent_gc

最新推荐文章于 2021-10-23 05:24:25 发布

阅读量397

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode

本文链接：https://blog.csdn.net/a529975125/article/details/79712643

Leetcode 专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一种使用动态规划解决特定类型数组操作问题的方法，旨在通过删除元素并获得相应分数的同时，确保不会删除相邻数值的元素。文章详细阐述了算法实现过程，并提供了两种不同版本的代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given an array nums of integers, you can perform operations on the array.

In each operation, you pick any nums[i] and delete it to earn nums[i] points. After, you must delete every element equal to nums[i] - 1 or nums[i] + 1.

You start with 0 points. Return the maximum number of points you can earn by applying such operations.
##思路
利用动态规划的思想，先用计数器（counter）算出nums中每个数的数量，得到一个dic他, 然后从最小数开始遍历：
1、当有两个相差为1的值相邻时，如（[2,2,3]），计算2×2和3的最大值，存到dict[2]和dict[3]中
2、当有两个相差大于1的值相邻时，如（[2,2,4]），则计算2×2+4，存到dict[4]中。

代码

version 1.

class Solution(object):
    def deleteAndEarn(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if not nums:
            return 0
        if len(nums) == 1:
            return nums[0]
        dict = collections.Counter(nums)
        sort_dict = sorted(dict.keys())
        dict[sort_dict[0]] = dict[sort_dict[0]] * sort_dict[0]
        if sort_dict[1] - sort_dict[0] == 1:
            dict[sort_dict[1]] = max(dict[sort_dict[1]] * sort_dict[1], dict[sort_dict[0]])
        else:
            dict[sort_dict[1]] = dict[sort_dict[1]] * sort_dict[1] + dict[sort_dict[0]]
        for i in range(2,len(sort_dict)):
            if sort_dict[i] - sort_dict[i-1] == 1:
                dict[sort_dict[i]] = max(dict[sort_dict[i]] * sort_dict[i] +dict[sort_dict[i-2]],
                                         dict[sort_dict[i-1]])
            else:
                dict[sort_dict[i]] = dict[sort_dict[i]] * sort_dict[i] +dict[sort_dict[i-1]]
        return max(dict.values())

第一版代码是直接在counter得到的字典里进行的操作。
第二版开辟了一个新数组来存挣到的pointers,快了一丢丢，大约5%的样子
version 2:

        if not nums:
            return 0
        if len(nums) == 1:
            return nums[0]
        dict = collections.Counter(nums)
        dp = [0] * (len(nums))
        sort_dict = sorted(dict.keys())
        dp[0] = dict[sort_dict[0]] * sort_dict[0]
        if sort_dict[1] - sort_dict[0] == 1:
            dp[1] = max(dict[sort_dict[1]] * sort_dict[1], dp[0])
        else:
            dp[1] = dict[sort_dict[1]] * sort_dict[1] + dp[0]
        for i in range(2,len(sort_dict)):
            if sort_dict[i] - sort_dict[i-1] == 1:
                dp[i] = max(dict[sort_dict[i]] * sort_dict[i] +dp[i-2],
                                         dp[i-1])
            else:
                dp[i] = dict[sort_dict[i]] * sort_dict[i] +dp[i-1]
        return max(dp)