【坚持每日一题5.5】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

思路：

动态规划，定义dp[i]表示到第i个元素所能得到的最大点数，唯一的坑点是注意去重！

java代码：

class Solution {
    public int deleteAndEarn(int[] nums) {
        
    	if(nums.length==0) return 0;
    	if(nums.length==1) return nums[0];
    	
    	Arrays.parallelSort(nums);
    	
    	int len=0;
    	int[] arr=new int[nums.length];
    	int[] sum=new int[100005];
    	boolean[] flag=new boolean[100005];
    	
    	for(int i=0;i<nums.length;i++) {
    		sum[nums[i]]++;
    		if(flag[nums[i]])
    			continue;
    		flag[nums[i]]=true;
    		arr[len++]=nums[i];
    	}
    	
    	int[] dp=new int[len];
    	
    	if(len==0) return 0;
    	if(len==1) return nums[0];
    	
    	dp[0]=arr[0]*sum[arr[0]];
    	dp[1]=(arr[1]==arr[0]+1)?Math.max(dp[0], arr[1]*sum[arr[1]]):arr[1]*sum[arr[1]]+arr[0]*sum[arr[0]];
    	for(int i=2;i<len;i++) {
    		if(arr[i]==arr[i-1]+1)
    			dp[i]=Math.max(dp[i-1], dp[i-2]+arr[i]*sum[arr[i]]);
    		else
    			dp[i]=dp[i-1]+arr[i]*sum[arr[i]];
    	}
    	
    	return dp[len-1];
    }
}