【Leetcode_总结】 740. 删除与获得点数 - python

Q:

给定一个整数数组 nums ，你可以对它进行一些操作。

每次操作中，选择任意一个 nums[i] ，删除它并获得 nums[i] 的点数。之后，你必须删除每个等于 nums[i] - 1 或 nums[i] + 1 的元素。

开始你拥有 0 个点数。返回你能通过这些操作获得的最大点数。

链接：https://leetcode-cn.com/problems/delete-and-earn/submissions/

思路：使用动态规划的方法解决问题，状态矩阵dp保存第i个值删除或者保留所获得的最大收益，状态转移矩阵为

dp[i] = max(dp[i - 1], i * dic[i] + dp[i - 2]) 。

代码：

class Solution(object):
    def deleteAndEarn(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if not nums:
            return 0
        dic = {}
        for num in nums:
            if num in dic:
                dic[num] += 1
            else:
                dic[num] = 1
        dp = [0 for _ in range(max(nums)+1)]
        if 1 in dic:
            dp[1] = max(dp[0], dic[1])
        else:
            dp[1] = 0
        for i in range(2,len(dp)):
            if i in dic:
                dp[i] = max(dp[i - 1], i * dic[i] + dp[i - 2])
            else:
                dp[i] = dp[i - 1]
        return max(dp)