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]范围内。
思路:网上很多解法都说和 House Robber很像,可以转化为这道题的解法,但是思考了很久也不知道该如何转化,主要是因为这里相邻的数字不一定就是nums[i]-1或者nums[i]+1,而且就算数组排序了也不能保证nums[i]前面或者后面的数字是nums[i]-1或者nums[i]+1,比如:
1,3,5

最后答案是1+3+5=9即所有数字都可以选。。。所以这道题要转化为House Robber的题必须要构造这样一种数列,满足不能连续偷相邻的数字的特性,观察到题目给出的nums[i]的范围不超过10000,所以我们把给的数组的每个数字映射到0~10000一共10001个数字的新的数组中,并且相同的数字累加到相同下标下,比如图例所示:

nums = [2, 2, 3, 3, 3, 4]

由于只含有数字2,3,4,所以最后映射的新数组sums中只有sums[2],sums[3],sums[4]不等于0,其余都等于0,并且sums[i]的更新规则如下:

for (int i : nums) sums[i] += i;

即是相同数字的累加和,即sums[2]=nums[0]+nums[1]=4,sums[3]=nums[2]+nums[3]+nums[4]=9,sums[4]=nums[5]=4

最后的sums等于:

[0,0,4,9,4,0......]

把相同数字压缩成和的形式是因为相同数字总是要么选要么不选的状态,即具有一致性。构建了新数组后我们就可以从0~10000根据House Robber的方法去解题了,定义两个变量take和skip表示上一次拿和不拿的最大价值,均初始化为0,每次更新是定义新的两个变量takei和skipi表示这次拿和这次不拿的最大价值,那么takei和skipi的更新公式如下:

int takei = skip + sums[i];       //这次拿的价值等于上次不拿的价值+第i件商品的价值
int skipi = max(take, skip);      //这次不拿的价值等于上次拿或者不拿的最大价值

最后返回take和skip的最大值

参考代码:

class Solution {
public:
    int deleteAndEarn(vector<int>& nums) {
	vector<int> sums(10001,0);
	for (int i : nums) sums[i] += i;
	int take = 0, skip = 0;
	for (int i = 0; i < 10001; i++) {
		int takei = skip + sums[i];
		int skipi = max(take, skip);
		take = takei;
		skip = skipi;
	}
	return max(take, skip);        
    }
};





猜你喜欢

转载自blog.csdn.net/qq_26410101/article/details/80896585