Week10 作业C - 拿数问题 II

1.题意

YJQ 上完第10周的程序设计思维与实践后，想到一个绝妙的主意，他对拿数问题做了一点小修改，使得这道题变成了 拿数问题 II。

给一个序列，里边有 n 个数，每一步能拿走一个数，比如拿第 i 个数， Ai = x，得到相应的分数 x，但拿掉这个 Ai 后，x+1 和 x-1 (如果有 Aj = x+1 或 Aj = x-1 存在) 就会变得不可拿（但是有 Aj = x 的话可以继续拿这个 x）。求最大分数。

2.样例

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

Hint

对于第三个样例：先选任何一个值为2的元素，最后数组内剩下4个2。然后4次选择2，最终得到10分。

3.解题思路

1. 这个与课上的题类似，只不过要进一步处理一下，课本上是Ai-1,Ai,Ai+1而本题要找x-1,x,x+1，所以我们可以把dp[i]的下标i替换为x即可
2. 本题还是要注意数据范围n*a[i]最大超过了int的数据范围
3. 思路比较难想，还是要多见变形题，多思考

4.AC代码

#include<iostream>
#include<algorithm>
using namespace std;
const int M=1e5+5;
long long a[M],f[M]={0};
long long dp[M]={0};
int main()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		f[a[i]]++;
	}
	sort(a+1,a+n+1);
	dp[0]=0;
	dp[1]=f[1]*1;
	long long mx=dp[1];
	for(int i=a[1];i<=a[n];i++)
	{
		if(i!=1)
		{
			dp[i]=max(dp[i-1],dp[i-2]+f[i]*i);
		}
		if(dp[i]>mx)
		{
			mx=dp[i];
		}
	}
	cout<<mx<<endl;
	return 0;
}