删除并获得点数
给你一个整数数组 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 个点数。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/delete-and-earn
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
我的思路:首先明确这样一个概念,每当选择一个点数x时,该元素以及所有等于 x-1或 x+1的元素会从数组中删去,而大于x+1(不包括x+1)以及小于x-1(不包括x-1)的点数不会受到任何牵连。为了方便理解,可以把nums数组转化一下,转化的目的是记录nums中每个点数出现的次数,我们用store数组来存储转化后的数组。转化后就可以用打家劫舍题目的思想解决这个题
class Solution {
public:
int deleteAndEarn(vector<int>& nums) {
if(nums.size()==0) return 0;
if(nums.size()==1) return nums[0];
//定义store数组,计数nums中每个点数出现的次数
vector<int> store = vector<int>(10001,0);
int MAX = nums[0];
//store数组的索引代表点数,数组元素值代表该点数出现的次数
for(int x:nums){
store[x]++;
MAX = max(MAX,x);
}
//定义dp数组
vector<int> dp(MAX+1);
//dp数组初始化
dp[0] = store[0]*0;//其实就是0
dp[1] = store[1]*1;//就是它本身
for(int i =2;i<=MAX;i++){
dp[i] = max(dp[i-1],dp[i-2]+store[i]*i);
}
return dp[MAX];
}
};
可以优化一下,当前位置的结果只与前两个结果有关,用滚动数组的思想压缩状态
class Solution {
public:
int deleteAndEarn(vector<int>& nums) {
if(nums.size()==0) return 0;
if(nums.size()==1) return nums[0];
//定义store数组,计数nums中每个点数出现的次数
vector<int> store = vector<int>(10001,0);
int MAX = nums[0];
//store数组的索引代表点数,数组元素值代表该点数出现的次数
for(int x:nums){
store[x]++;
MAX = max(MAX,x);
}
//定义dp数组,滚动数组思想
int first = 0;
int second = store[1];
for(int i =2;i<=MAX;i++){
int temp = second;
second = max(first+store[i]*i,second);
first = temp;
}
return second;
}
};