Skip to content

Files

Latest commit

f05a156 · Apr 18, 2019

History

History
 
 

n0740. Delete and Earn

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 

Delete and Earn ⭐⭐

题目内容

给定一个整数数组 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]范围内。

解法

// Author: Netcan @ https://github.com/netcan/Leetcode-Rust
// Zhihu: https://www.zhihu.com/people/netcan

impl Solution {
    pub fn delete_and_earn(nums: Vec<i32>) -> i32 {
        // 打家劫舍问题
        let mut gold = [0; 10001];
        for &n in &nums {
            gold[n as usize] += n;
        }
        let mut dp = [0; 10001];
        let (mut take, mut max_earn) = (0, 0);
        for i in 1..=10000usize {
            dp[i] = if i > 1 {
                take = take.max(dp[i - 2]); // dp[0..=i-2].max()
                take + gold[i]
            } else {
                gold[i]
            };
            max_earn = max_earn.max(dp[i]);
        }

        max_earn
    }
}