力扣 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 个点数。

提示：

• 1 <= nums.length <= 2 * 10^4
• 1 <= nums[i] <= 10^4

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/delete-and-earn
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

动态规划

首先分析一下题目中的操作，只要选择了某个数字 num，在获得删除它并获得相应的分值之后，同时会删除数组中的所有 num - 1 以及 num + 1。为了保证得分最高，对于剩下来的那些 num，必须要把它们选完，而且在后续选择过程中，不用删除 num - 1 以及 num + 1，因为在第一次操作过程中已经删除完了。

所以对于某个数字 num，如果要选择它，那么对应的最大分值就是 num * c，其中 c 为 num 在 nums 数组中出现的次数。在 Python 语言中，可以借助 collections.Counter 来轻松完成计数工作，它本质上是一个字典。

另外，数组中元素顺序对于结果没有影响，如果我们按照从小到大的顺序考虑某个数字 num 选或者不选，那么不需要考虑 num + 1，因为在升序条件下， num + 1 还未出现。

对于数组中没有出现过的数字，为了应用状态转移方程，可以考虑成出现为 0 次，这样就不算分了。这与 collections.Counter 的工作方式一致，如果访问 Counter 中不存在的元素，会返回 0，而不会抛出 KeyError 异常。

状态定义：f[i] 为数组中数字 1 ~ i 通过操作能获取的最大分值。

状态转移：

• 对于数字 i，如果选（即删除并获取分值），得分为 f[i - 2] + i * c[i]
• 对于数字 i，如果不选（即放弃操作），的氛围 f[i - 1]
• 两者取最大即可

class Solution: