【leetcode】740. Delete and Earn

题目描述：

 

题目大意：

通过对一个整数数组中任意一个元素进行一种操作，可以得到分数，这种操作的过程为：

取出一个元素a，得到分数a，同时删去这个数组中所有比该元素小1或者大1的数。

求所能得到的分数最大值。

解题方法：

这道题给出了数据范围在e4，基本确定算法复杂度不会超过nlogn。因此使用dfs 回溯的几率很小，优先考虑动态规划（遇事不决动态规划）

首先对nums进行排序

之后定义形状为（len(nums),2）的dp数组

dp[i][0]表示保留nums[i]后，在nums[:i]区间内可以得到的分数最大值。

dp[i][1]表示舍弃nums[i]后，在nums[:i]区间内可以得到的分数最大值

状态转移方程为：

if nums[i]==nums[i-1]:
    dp[i][0] = dp[i-1][0]+nums[i]
    dp[i][1] = dp[i-1][1]
elif nums[i]==nums[i-1]+1:
    dp[i][0] = dp[i-1][1]+nums[i]
    dp[i][1] = max(dp[i-1][0],dp[i-1][1])
else:
    dp[i][0] = max(dp[i-1][0],dp[i-1][1])+nums[i]
    dp[i][1] = max(dp[i-1][0],dp[i-1][1])

python代码为：

class Solution(object):
    def deleteAndEarn(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        
        nums = sorted(nums)
        if len(nums)==1:
            return nums[0]
        dp = [[0,0] for _ in range(len(nums))]
        res = 0
        dp[0][0] = nums[0]
        for i in range(1,len(nums)):
            if nums[i]==nums[i-1]:
                dp[i][0] = dp[i-1][0]+nums[i]
                dp[i][1] = dp[i-1][1]
            elif nums[i]==nums[i-1]+1:
                dp[i][0] = dp[i-1][1]+nums[i]
                dp[i][1] = max(dp[i-1][0],dp[i-1][1])
            else:
                dp[i][0] = max(dp[i-1][0],dp[i-1][1])+nums[i]
                dp[i][1] = max(dp[i-1][0],dp[i-1][1])
            res = max(res,dp[i][0],dp[i][1])
        

        
        return res

时间复杂度为O（nlongn）

且dp操作部分时间复杂度为O（n）

题目中虽然要求nums[i]选中后，比其小1和大1的全都要舍弃，但我们在填写dp数组时只考虑与比nums[i]小1的值带来的影响。

之所以依然能够得到解一是因为我们首先对nums进行了排序，二是因为我们对dp数值的定义：dp[i]只考虑nums[:i]的解，因此不需要考虑比nums[i]大1的元素。

解法正确性可通过数学归纳法进行证明：

当n=1时，算法正确，

我们假设当n=k时，算法正确，

当n=k+1时，经过dp状态转移方程计算后的dp[k+1]的值符合dp定义，算法依然正确。

由数学归纳法可知，算法正确性得证。