难度:中等
给你一个整数数组
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 * 104
1 <= nums[i] <= 104
分析:
动态规划题,遍历,当前值只有两种选择:
①放弃当前点数,到现在为止的最大点数为前一个下标(i-1)的最大点数;
②获取当前点数, 前两个下标(i-2)的最大点数+当前点数。
两者取max即可。
原始nums数组,存在重复值且无序,需要先处理为一维数组,下标为对应的值,存储的值为值的个数count。
代码:
class Solution {
public int deleteAndEarn(int[] nums) {
int len = nums.length;
// find max element
int max = 0;
for (int i = 0; i < len; i++) {
max = Math.max(max, nums[i]);
}
// create new array to store element and count
int[] countNums = new int[max + 1];
for (int i = 0; i < len; i++) {
countNums[nums[i]]++;
}
//create array to record the best resolution of every index
int[] dp = new int[max + 1];
//init
dp[0] = 0;
dp[1] = countNums[1];
//calculate
for (int i = 2; i < countNums.length; i++) {
dp[i] = Math.max(dp[i - 1], dp[i - 2] + i * countNums[i]);
}
return dp[max];
}
}