week10 作业C 拿数问题II

拿数问题 II

给 n 个数，每一步能拿走一个数，比如拿第 i 个数， Ai = x，得到相应的分数 x，但拿掉这个 Ai 后，x+1 和 x-1 (如果有 Aj = x+1 或 Aj = x-1 存在) 就会变得不可拿（但是有 Aj = x 的话可以继续拿这个 x）。求最大分数。
本题和课上讲的有些许不一样，但是核心是一样，需要你自己思考。

Input

第一行包含一个整数 n (1 ≤ n ≤ 105)，表示数字里的元素的个数
第二行包含n个整数a1, a2, …, an (1 ≤ ai ≤ 105)

Output

输出一个整数：n你能得到最大分值。

Example

Input
2
1 2
Output
2
Input
3
1 2 3
Output
4
Input
9
1 2 1 3 2 2 2 2 3
Output
10

我的思路：
这道题与课堂上讲的不同，课堂是位置相邻的不能加，而这道题是在分数大小上相邻的不能加而相同分数可以加。我们可以根据这个特点，对于问题进行一个转化，可以记录
每个数字的出现次数，从而形成一个数组sum，索引值为这个数字，里面的数即为出现的次数，这样就把问题又转变成了相邻不能加的问题。于是，可以推出从当前位置到最前方的最大分数dp[i] = max(dp[i-1],dp[i-2]+i*sum[i])。这里可以先输入的时候，记录所有数字的最小值minx和最大值maxx，然后最后做dp递推时，直接从minx到maxx计算即可。
我的总结：
最开始，我的计算方法是对于这些数字进行升序排序，然后遍历找之前比自己的小2及以上或等于自己的数字加上他们的dp作为当前数的dp,最后找到最大的dp即可。然而，这种方法较为朴素，时间复杂度太高，会TLE，所以最后选择上文所说的方法。

我的代码：

#include<iostream>
#include<algorithm>
using namespace std;

long  long A[100010],dp[100010],sum[100010],n,minx=100010,maxx=0;

int main()
{
	cin>>n;
	long  long ans=0;
	for(long  long i=1;i<=n;i++) 
	{
		cin>>A[i];
		sum[A[i]]++;
		minx = min(minx,A[i]);
		maxx = max(maxx,A[i]);
	}
	for(int i=minx;i<=maxx;i++)
	{
		dp[i] = max(dp[i-1],dp[i-2]+i*sum[i]);
		ans = max(ans,dp[i]);
	}
	cout<<ans;
	return 0;
}