题目描述:
给你一个整数数组 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
分析
统计每个点数出现的次数,如果选择一个点数的话,肯定是把所有的这个点数都选择下来的情况是最优的,同时对点数排序,如果两个点数之间的差距大于1,那么这两个点数都能选,如果不大于1,他就看是去当前点数加上上上次取得点数打,还是上一次取得点数大。
dp[i+1] = Math.max(dp[i], dp[i-1] + map.get(arr[i]) * arr[i])
(dp[i]表示在排序后的数组中的i位置的最大值)
代码:
/**
* @param {number[]} nums
* @return {number}
*/
var deleteAndEarn = function(nums) {
let len = nums.length;
const arr = [];
const map = new Map();
for(let i = 0; i < len; i += 1) {
if(map.has(nums[i])) {
map.set(nums[i], map.get(nums[i]) +1);
} else {
map.set(nums[i], 1);
arr.push(nums[i]);
}
}
arr.sort((a, b) => {return a - b});
const dp = new Array(len).fill(0);
dp[0] = 0;
dp[1] = map.get(arr[0]) * arr[0];
len = arr.length
for(let i = 1; i < len; i += 1) {
if(arr[i] - arr[i-1] > 1)
dp[i+1] = dp[i] + map.get(arr[i]) * arr[i];
else {
dp[i+1] = Math.max(dp[i], dp[i-1] + map.get(arr[i]) * arr[i]);
}
}
return dp[len];
};