题目:删除并获得点数
给你一个整数数组 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 个点数。
题目 " 删除并获得点数" 的关键在于如何选择删除某个数字,并且获得该数字对应的点数,同时避免被强制删除相邻的数字。你的思路是对的,但代码还有一些需要改进的地方。让我帮你进一步梳理解题思路,并给出一个基于你当前思路的解决方案。
解题思路:
-
统计每个数字的总分数:
- 我们不关心某个数字具体出现了多少次,而是更关心它给我们提供的总分数。因此,首先你要统计每个数字
i
在数组中的总贡献值(即count[i] * i
)。
- 我们不关心某个数字具体出现了多少次,而是更关心它给我们提供的总分数。因此,首先你要统计每个数字
-
类似于打家劫舍问题:
- 这道题类似于“打家劫舍”问题,即你不能同时选择相邻的数字。你要在每次选择一个数字
i
时,决定是否选择它,并相应地选择或放弃i-1
和i+1
。
- 这道题类似于“打家劫舍”问题,即你不能同时选择相邻的数字。你要在每次选择一个数字
-
动态规划的思想:
- 动态规划数组
dp[i]
用来表示选择数字i
时所能获得的最大点数。 - 状态转移:对于每个数字
i
,你有两个选择:- 要么选择
i
,并加上dp[i-2]
的最大点数(因为i-1
不能选)。 - 要么放弃
i
,则dp[i] = dp[i-1]
。
- 要么选择
- 动态规划数组
class Solution(object):
def deleteAndEarn(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
if not nums:
return 0
# 1. 统计每个元素的总点数
count = {}
for num in nums:
if num in count:
count[num] += num
else:
count[num] = num
# 2. 对于出现过的数字,我们用一个 dp 数组来存储可以获得的最大点数
# 将所有唯一的数字排序,以确保我们能正确地处理相邻的数字
unique_nums = sorted(count.keys())
# 3. 动态规划数组初始化
n = len(unique_nums)
dp = [0] * (n)
# 4. 初始化 dp[0](处理第一个数字)
dp[0] = count[unique_nums[0]]
# 5. 从第二个数字开始递推
for i in range(1, n):
num = unique_nums[i]
if unique_nums[i] == unique_nums[i - 1] + 1:
# 如果当前数字和前一个数字相邻
dp[i] = max(dp[i - 1], dp[i - 2] + count[num] if i - 2 >= 0 else count[num])
else:
# 如果当前数字和前一个数字不相邻
dp[i] = dp[i - 1] + count[num]
return dp[-1] # 最后一个元素的值就是我们能够获得的最大点数