数据结构与算法-动态规划-删除并获得点数

删除并获得点数

给你一个整数数组 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 个点数。

思路

根据题意，在选择了元素 x 后，该元素以及所有等于 x−1 或 x+1 的元素会从数组中删去。若还有多个值为 x 的元素，由于所有等于 x−1 或 x+1 的元素已经被删除，我们可以直接删除 x 并获得其点数。因此若选择了 x，所有等于 x 的元素也应一同被选择，以尽可能多地获得点数。

记元素 x 在数组中出现的次数为 c，我们可以用一个数组 sum 记录数组 nums 中所有相同元素之和，即 sum[x]=x⋅c。若选择了 x，则可以获取 sum[x] 的点数，且无法再选择 x−1 和 x+1。这与「 打家劫舍」是一样的。

代码：

class Solution {
      public int deleteAndEarn(int[] nums) {
        int maxVal = 0;
        for (int num : nums) {
            maxVal = Math.max(maxVal, num);
        }
        int[] sums = new int[maxVal+1];
        for (int num : nums) {
            sums[num] += num;
        }
        return rob(sums);
    }

    private int rob(int[] nums) {
        int size = nums.length;
        int first = nums[0];
        int second = Math.max(first, nums[1]);
        for (int i = 2; i < size; i++) {
            int temp = second;
            second = Math.max(first + nums[i], second);
            first = temp;
        }
        return second;
    }
}