给你一个整数数组 nums ,你可以对它进行一些操作。
每次操作中,选择任意一个 nums[i] ,删除它并获得 nums[i] 的点数。之后,你必须删除 所有 等于 nums[i] - 1 和 nums[i] + 1 的元素。
开始你拥有 0 个点数。返回你能通过这些操作获得的最大点数。
来源:力扣(LeetCode)
题目链接:https://leetcode-cn.com/problems/delete-and-earn
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题方法一:动态规划
class Solution {
public:
int deleteAndEarn(vector<int>& nums) {
int max_val = 0;
for(int num : nums) max_val = max(num, max_val);
vector<int> dp(max_val + 1); //dp[i] 为 nums 中值为 i 的数字的个数
dp[0] = 0;
for(int i = 0; i < nums.size(); ++i){
dp[nums[i]]++;
}
for(int i = 2; i <= max_val; ++i){
dp[i] = max(dp[i - 2] + dp[i] * i, dp[i - 1]);//更新dp数组,让其为题目结果
}
return dp[max_val];
}
};
解题方法二:动规 + 哈希表
class Solution {
public:
int deleteAndEarn(vector<int>& nums) {
sort(nums.begin(), nums.end());
unordered_map<int, int> umap;
vector<int> dp = {0, nums[0]}; //dp数组存储从小到大的nums数组中的值(不重复)
umap[nums[0]]++;
for(int i = 1; i < nums.size(); ++i){
umap[nums[i]]++;
if(nums[i] != dp.back()) dp.push_back(nums[i]);
}
int last = dp[1];
dp[1] = dp[1] * umap[dp[1]];
for(int i = 2; i < dp.size(); ++i){
if(dp[i] == last + 1){
last = dp[i];
dp[i] = max(dp[i - 1], dp[i - 2] + dp[i] * umap[dp[i]]);
}
else{
last = dp[i];
dp[i] = dp[i - 1] + dp[i] * umap[dp[i]];
}
}
return dp.back();
}
};