JAVA程序设计：删除与获得点数（LeetCode：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 个点数。
注意:

nums的长度最大为20000。
每个整数nums[i]的大小都在[1, 10000]范围内。

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

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];
    }
}