【动态规划】删除并获得点数

删除并获得点数

给你一个整数数组 nums ，你可以对它进行一些操作。

每次操作中，选择任意一个 nums[i] ，删除它并获得 nums[i] 的点数。之后，你必须删除 所有 等于 nums[i] - 1 和 nums[i] + 1 的元素。

开始你拥有 0 个点数。返回你能通过这些操作获得的最大点数。

示例 1：

输入：nums = [3,4,2]

输出：6

解释：

删除 4 获得 4 个点数，因此 3 也被删除。

之后，删除 2 获得 2 个点数。总共获得 6 个点数。

示例 2：

输入：nums = [2,2,3,3,3,4]

输出：9

解释：

删除 3 获得 3 个点数，接着要删除两个 2 和 4 。

之后，再次删除 3 获得 3 个点数，再次删除 3 获得 3 个点数。

总共获得 9 个点数。

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/delete-and-earn

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

我的思路：首先明确这样一个概念，每当选择一个点数x时，该元素以及所有等于 x-1或 x+1的元素会从数组中删去，而大于x+1（不包括x+1）以及小于x-1（不包括x-1）的点数不会受到任何牵连。为了方便理解，可以把nums数组转化一下，转化的目的是记录nums中每个点数出现的次数，我们用store数组来存储转化后的数组。转化后就可以用打家劫舍题目的思想解决这个题

	class Solution {
public:
    int deleteAndEarn(vector<int>& nums) {
        if(nums.size()==0) return 0;
        if(nums.size()==1) return nums[0];
        //定义store数组，计数nums中每个点数出现的次数
        vector<int> store = vector<int>(10001,0);
        int MAX = nums[0];
        //store数组的索引代表点数，数组元素值代表该点数出现的次数
        for(int x:nums){
            store[x]++;
            MAX = max(MAX,x);
        }
        //定义dp数组
        vector<int> dp(MAX+1);
        //dp数组初始化
        dp[0] = store[0]*0;//其实就是0
        dp[1] = store[1]*1;//就是它本身
        for(int i =2;i<=MAX;i++){
            dp[i] = max(dp[i-1],dp[i-2]+store[i]*i);
        }
        return dp[MAX];
    }
};

可以优化一下，当前位置的结果只与前两个结果有关，用滚动数组的思想压缩状态

class Solution {
public:
    int deleteAndEarn(vector<int>& nums) {
        if(nums.size()==0) return 0;
        if(nums.size()==1) return nums[0];
        //定义store数组，计数nums中每个点数出现的次数
        vector<int> store = vector<int>(10001,0);
        int MAX = nums[0];
        //store数组的索引代表点数，数组元素值代表该点数出现的次数
        for(int x:nums){
            store[x]++;
            MAX = max(MAX,x);
        }
        //定义dp数组，滚动数组思想
        int first = 0;
        int second = store[1];
        for(int i =2;i<=MAX;i++){
            int temp = second;
            second = max(first+store[i]*i,second);
            first = temp;
        }
        return second;
    }
};