题目链接
题目描述
给你一个整数数组 n u m s nums nums ,你可以对它进行一些操作。
每次操作中,选择任意一个 n u m s [ i ] nums[i] nums[i] ,删除它并获得 n u m s [ i ] nums[i] nums[i] 的点数。之后,你必须删除 所有 等于 n u m s [ i ] − 1 nums[i] - 1 nums[i]−1 和 n u m s [ i ] + 1 nums[i] + 1 nums[i]+1 的元素。
开始你拥有 0 0 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 个点数。
提示:
- 1 ≤ n u m s . l e n g t h ≤ 2 ∗ 1 0 4 1 \leq nums.length \leq 2 * 10^4 1≤nums.length≤2∗104
- 1 ≤ n u m s [ i ] ≤ 1 0 4 1 \leq nums[i] \leq 10^4 1≤nums[i]≤104
解法:动态规划
我们用 c n t [ x ] cnt[x] cnt[x] 表示, n u m s nums nums 中元素 x x x 的数量。
我们如果选择了 x x x,那么所能获得的点数就为 x × c n t [ x ] x \times cnt[x] x×cnt[x]。既然选择了 x x x,那么就要删除 x − 1 x - 1 x−1 和 x + 1 x + 1 x+1,删除就等同于 不能选 x − 1 x-1 x−1 和 x + 1 x+1 x+1。
所以这个实际上是一个 打家劫舍 的变形题。
关于打家劫舍的解决方法:Leetcode.198 打家劫舍
时间复杂度: O ( n ) O(n) O(n)
C++代码:
class Solution {
public:
int deleteAndEarn(vector<int>& nums) {
int mx = 0;
for(auto x:nums) mx = max(mx , x);
vector<int> cnt(mx + 1);
for(auto x:nums) cnt[x]++;
int f0 = 0 , f1 = 0;
for(int i = 1;i <= mx;i++){
int new_f = max(f0 + i * cnt[i] , f1);
f0 = f1;
f1 = new_f;
}
return f1;
}
};