LWC 61:740. Delete and Earn

LWC 61:740. Delete and Earn

传送门:740. Delete and Earn

Problem:

Given an array nums of integers, you can perform operations on the array. In each operation, you pick any nums[i] and delete it to earn nums[i] points. After, you must delete every element equal to nums[i] - 1 or nums[i] + 1. You start with 0 points. Return the maximum number of points you can earn by applying such operations.

Example 1:

Input: nums = [3, 4, 2] Output: 6 Explanation: Delete 4 to earn 4 points, consequently 3 is also deleted. Then, delete 2 to earn 2 points. 6 total points are earned.

Example 2:

Input: nums = [2, 2, 3, 3, 3, 4] Output: 9 Explanation: Delete 3 to earn 3 points, deleting both 2’s and the 4. Then, delete 3 again to earn 3 points, and 3 again to earn 3 points. 9 total points are earned.

Note:

The length of nums is at most 20000.

Each element nums[i] is an integer in the range [1, 10000].

思路: 最简单的思路,选择一个数后,比如[3, 4, 2]选择4,那么3失效,剩余数组为[2],当作子问题来求解即可,这是一种情况。选择3,那么4和2均失效,剩余数组为[],也可以当作子问题求解。选择2,则3失效,剩余数组为[4],在这几种情况中选择profit最多的情况即可。

但上述递归存在一个严重的问题,子问题无法被记录下来,因为选择是无序的。因此可以采用背包的思路,先对这些元素排序,这样数组变为[2, 3, 4],这样一来对于2选or不选,进一步影响3是否被包含,子问题能够轻而易举的被状态化,因为此时当前位置i,那么对应的i到n的位置可以表示子问题。而这些子问题与第一种求解思路不同的是,它们因位置而存在唯一性。

Java版本:(递归+记忆化)

    public int deleteAndEarn(int[] nums) {
        Map<Integer, Integer> mem = new HashMap<>();
        for (int n : nums) {
            mem.put(n, mem.getOrDefault(n, 0) + 1);
        }

        int[] cnt = new int[mem.size()];
        Set<Integer> keys = mem.keySet();

        int[] key = keys.stream().mapToInt(i -> i).toArray();
        Arrays.sort(key);

        for (int i = 0; i < key.length; ++i) {
            cnt[i] = mem.get(key[i]);
        }
        dp = new int[20000 + 16];
        return robot(key, cnt, 0);
    }

    int[] dp;
    public int robot(int[] key, int[] cnt, int pos) {
        if (pos >= key.length) return 0;
        if (dp[pos] > 0) return dp[pos];

        int ans = 0;
        int choose = 0;

        // choose
        if (pos + 1 < key.length && key[pos + 1] - 1 == key[pos]) {
            choose = key[pos] * cnt[pos] + robot(key, cnt, pos + 2);
        }
        else {
            choose = key[pos] * cnt[pos] + robot(key, cnt, pos + 1);
        }
        ans = Math.max(ans, choose);
        // no choose
        ans = Math.max(ans, robot(key, cnt, pos + 1));
        return dp[pos] = ans;
    }

递推版本:

    public int deleteAndEarn(int[] nums) {
        if (nums.length == 0) return 0;

        Map<Integer, Integer> mem = new HashMap<>();
        for (int n : nums) {
            mem.put(n, mem.getOrDefault(n, 0) + 1);
        }

        int[] cnt = new int[mem.size()];
        Set<Integer> keys = mem.keySet();

        int[] key = keys.stream().mapToInt(i -> i).toArray();
        Arrays.sort(key);

        for (int i = 0; i < key.length; ++i) {
            cnt[i] = mem.get(key[i]);
        }

        int[] dp = new int[20000 + 16];
        int last = key.length;

        dp[last - 1] = key[last - 1] * cnt[last - 1];
        for (int i = last - 2; i >= 0; --i) {
            if (key[i] + 1 == key[i + 1]) {
                dp[i] = Math.max(dp[i], dp[i + 2] + key[i] * cnt[i]);
            }
            else {
                dp[i] = Math.max(dp[i], dp[i + 1] + key[i] * cnt[i]);
            }

            dp[i] = Math.max(dp[i], dp[i + 1]);
        }

        return dp[0];
    }

简化map版本,用数组位置排序+计数:

      public int deleteAndEarn(int[] nums) {
          int[] cnt = new int[10016];
          for (int num : nums) cnt[num]++;

          int[] dp = new int[10016];
          dp[0] = 0;
          dp[1] = cnt[1];

          for (int i = 2; i <= 10000; ++i) {
              dp[i] = Math.max(dp[i - 1], dp[i - 2] + cnt[i] * i);
          }

          return dp[10000];
      }

Python版本:

    def deleteAndEarn(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        cnt = [0] * 10016
        for num in nums:
            cnt[num] = cnt[num] + 1
        dp = [0] * 10016
        dp[0] = 0
        dp[1] = cnt[1]

        for i in range(2, 10001):
            dp[i] = max(dp[i - 1], dp[i - 2] + i * cnt[i])

        return dp[10000]

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏NetCore

url带中文参数显示乱码的问题

最近要上个项目,其实很简单的东西,就是拼接一个url,不过url中的参数需要UrlEncode编码的,其实对我来说,这个问题很好解决,C#用HttpUtilit...

2039
来自专栏葡萄城控件技术团队

Spread for Windows Forms快速入门(14)---文件操作

你可以将数据从Spread中存到几种不同类型的文件中或者可以在Spread中打开几种不同类型的数据文件。通过使用相应的代码,你可以将整个控件,某个特定的表单,或...

1776
来自专栏DeveWork

【译】WordPress 中的50个过滤器(3):第11-20个过滤器

本文为系列第三篇,原文:50 Filters of WordPress: Filters 11-20 原文地址 不多说,直接进入正题。 本系列文章翻译自tut...

1986
来自专栏林德熙的博客

控件

Windows的XAML UI框架提供了很多控件,支持用户界面开发库。其中一些有可视化,一些布局。 一些控件例子:https://github.com/Micr...

521
来自专栏编程微刊

表格插件-bootstrap table的分页使用示例

4:完整的添加数据之后显示在表格里面,自动分页代码,删除和编辑效果 jsp界面 :

672
来自专栏数据科学与人工智能

IPython使用简易教程

本文总结Ipython使用的知识。 ? 1 IPython 是什么? IPython (short for Interactive Python) was st...

3124
来自专栏Python

linux每日命令(12):nl命令

nl命令在linux系统中用来计算文件中行号。nl 可以将输出的文件内容自动的加上行号!其默认的结果与 cat -n 有点不太一样, nl 可以将行号做比较多的...

692
来自专栏hotqin888的专栏

beego利用casbin进行权限管理——第五节 策略更新(续)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/hotqin888/article/det...

931
来自专栏Material Design组件

Human Interface Guidelines — Text Fields

1315
来自专栏游戏杂谈

测试Live Writer写日志

672

扫码关注云+社区