LeetCode-【动态规划】-删除与获得点数

版权声明:转载注明出处就可以了。。。 https://blog.csdn.net/zw159357/article/details/82593284

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

注意:

  • nums的长度最大为20000
  • 每个整数nums[i]的大小都在[1, 10000]范围内。

题解:这道题目其实与打家劫舍相似,如果你偷了某一家,那么与这一家相邻的两家就不能再偷,只不过在这里的表达变为选择任意一个 nums[i] ,删除它并获得 nums[i] 的点数。之后,你必须删除每个等于 nums[i] - 1 或 nums[i] + 1 的元素,删除nums[i] - 1 或 nums[i] + 1代表不偷相邻两家,问题似乎变得简单一些了,那接下来就是寻找题目的动态转移方程了。

参考题解:https://blog.csdn.net/OneDeveloper/article/details/80001202

class Solution {
    public int deleteAndEarn(int[] nums) {
        int[] sum=new int[10001];
        int n=nums.length;
        if(n==0)
            return 0;
        int max=0;
        for(int i=0;i<n;i++){
            sum[nums[i]]++;
            max=Math.max(max,nums[i]);
        }
        int[] dp=new int[max+1];
        dp[0]=sum[0]*0;
        dp[1]=sum[1]*1;
        for(int i=2;i<=max;i++){
            dp[i]=Math.max(dp[i-1],dp[i-2]+sum[i]*i);
        }
        return dp[max];
    }
}

猜你喜欢

转载自blog.csdn.net/zw159357/article/details/82593284