给你一个整数数组 nums ,你可以对它进行一些操作。
每次操作中,选择任意一个 nums[i] ,删除它并获得 nums[i] 的点数。之后,你必须删除 所有 等于 nums[i] - 1 和 nums[i] + 1 的元素。
开始你拥有 0 个点数。返回你能通过这些操作获得的最大点数。
示例 1:
输入:nums = [3,4,2]
输出:6
解释:
删除 4 获得 4 个点数,因此 3 也被删除。
之后,删除 2 获得 2 个点数。总共获得 6 个点数。
挺恶心的一道题,看了半天也才差不多知道了
主要注意for循环的结束点,有点搞头
class Solution {
public:
int deleteAndEarn(vector<int>& nums) {
//将问题变型(打家劫舍)
int max_num = *max_element(nums.begin(), nums.end());
vector<int>store(max_num+1, 0);
for(auto num: nums){
store[num]++;
}
//动态规划
int len = store.size();
vector<int>dp(len, 0);
dp[1] = store[1];
for(int i = 2; i < len; i++){
dp[i] = max(dp[i-1], dp[i-2] + store[i] * i);
}
return dp[len-1];
}
};
心态已经炸了