题目如下
数据范围
本题和背包问题很像:对于每种数字都只有装入或者不装入两种状态,都想要得到最值。
与一般的背包问题相比本题没有空间限制而是多了物品装入规则即放了3号物品就不能放入2号和4号物品所以我们可以仿照背包问题的思路讨论:
设f(0,i)是不放入i号物品得到的最大值 也就是说我们不删除值为i的数(这里的i是值而非数组索引)
设f(1,i)是放入i号物品得到的最大值
那么我们可知
f(0,i) = max(f(1,i - 1),f(0,i - 1))
f(1,i) = f(0,i - 1) + i * (i 的个数)
所以我们只需要得到num数组最大值以及各个数的个数即可。
显然在处理i的时候我们只需要i - 1的状态所以利用滚动数组的思想设置pre1 pre2 t三个变量
其中pre1 是i - 1的不放入状态 pre2是i - 1的放入状态 t暂存pre1(因为pre1会被改变所以需要暂存)。
通过代码
class Solution {
public:
int deleteAndEarn(vector<int>& nums) {
unordered_map<int,int> map;
int n = nums.size();
int max1 = -1;
for(int i = 0;i < n;i++) {
map[nums[i]] = map[nums[i]] + 1;
max1 = max(max1,nums[i]);
}
int pre1 = 0,pre2 = 1 * map[1];
int t;
for(int i = 2;i <= max1;i++) {
t = pre1;
pre1 = max(pre1,pre2);
pre2 = t + i * map[i];
}
return pre1 > pre2?pre1:pre2;
}
};