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 条评论
登录 后参与评论

相关文章

来自专栏小勇DW3

concrrent类下 BlockingDeque 下 自己实现代码编写

  java6增加了两种容器类型,Deque和BlockingDeque,它们分别对Queue和BlockingQueue进行了扩展。 Deque是一个双端队...

732
来自专栏机器学习入门

LWC 52:686. Repeated String Match

LWC 52:686. Repeated String Match 传送门:686. Repeated String Match Problem: Given...

1965
来自专栏Java学习网

Java中处理正则表达式的工具类——总有一个适合你

import java.util.ArrayList; import java.util.List; import java.util.regex.Match...

2685
来自专栏LinkedBear的个人空间

唠唠SE的集合-05——LinkedList 原

由于LinkedList:每一个元素在内存中的地址不连续,需要让上一个元素记住下一个元素。

771
来自专栏专注 Java 基础分享

详解Java动态代理机制(二)----cglib实现动态代理

     上篇文章的结尾我们介绍了普通的jdk实现动态代理的主要不足在于:它只能代理实现了接口的类,如果一个类没有继承于任何的接口,那么就不能代理该类,原因是我...

24010
来自专栏文武兼修ing——机器学习与IC设计

栈与栈的实现栈栈的基本操作栈的实现

栈 栈是一种基础的数据结构,只从一端读写数据。基本特点就”后进先出“,例如顺序入栈1,2,3,4,5,再顺序出栈是5,4,3,2,1 栈的基本操作 栈的基本操作...

2905
来自专栏zhisheng

#每日一题#3

如下代码,执行test()函数后,屏幕打印结果为() public class Test2 { public void add(Byte b) ...

36011
来自专栏一枝花算不算浪漫

一道笔试题来理顺Java中的值传递和引用传递

36811
来自专栏不想当开发的产品不是好测试

java array to list

背景 想把数组转为list,使用list的判断元素是否存在的方法,结果发现一个坑,int类型的数组失败了  步骤 public static void main...

1885
来自专栏老马说编程

(70) 原子变量和CAS / 计算机程序的思维逻辑

从本节开始,我们探讨Java并发工具包java.util.concurrent中的内容,本节先介绍最基本的原子变量及其背后的原理和思维。 原子变量 什么是原子...

1789

扫码关注云+社区