给定一个整数数组 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];
}
}