题目出处:https://leetcode-cn.com/problems/delete-and-earn/
思路:构造数组sum,表示删除指定元素i所能获得的点数是sum[i]。如果选择第sum[i]个数,则此时sum[i-1]和sum[i+1]都是不能选择的,问题就转化成了打家劫舍问题。
class Solution {
public:
int rob(vector<int>& Sum){ //打家劫舍
int n = Sum.size();
vector<int> dp(n, 0); //到达第i家时能获得的最大金额
dp[0] = Sum[0];
dp[1] = max(Sum[0], Sum[1]);
for(int i = 2; i < n; ++i){
dp[i] = max(dp[i-1], dp[i-2]+Sum[i]);
}
return dp[n-1];
}
int deleteAndEarn(vector<int>& nums) {
int max_ele = *max_element(nums.begin(), nums.end());
vector<int> Sum(max_ele+1);
for(auto val : nums){
Sum[val] += val;
}
return rob(Sum);
}
};```