题目
给你一个整数数组 nums
,你可以对它进行一些操作。
每次操作中,选择任意一个 nums[i]
,删除它并获得 nums[i]
的点数。之后,你必须删除 所有 等于 nums[i] - 1
和 nums[i] + 1
的元素。
开始你拥有 0 个点数。返回你能通过这些操作获得的最大点数。
题解 动态规划
分析一下,其实类似于打家劫舍这道题,附链接:
链接: https://blog.csdn.net/zzh123353/article/details/119809150?spm=1001.2014.3001.5501.
仔细分析,实际就是在选择一个元素x
以后,x + 1 与 x - 1
都无选择,所有x元素都会被选择,所以统计每个相同元素的总和放在另一个容器中,每个下标对应这个元素值的累加总和,就完全转化为了打家劫舍的问题,转移方程:
dp[i] = max(dp[i - 1], dp[i - 2] + nums[i]);
class Solution {
public:
int deleteAndEarn(vector<int>& nums) {
int MaxLen = INT_MIN;
for(auto num : nums){
MaxLen = max(MaxLen, num);
}
vector<int> res(MaxLen + 1, 0);
for(auto num:nums)
res[num] += num;
vector<int> dp(MaxLen + 1, 0);
dp[0] = res[0];
dp[1] = res[1];
for(int i = 2; i <= MaxLen; ++i){
dp[i] = max(dp[i - 1], dp[i - 2] + res[i]);
}
return dp[MaxLen];
}
};