week10作业 C - 拿数问题 II

题意

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

Hint

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

思路

本题选取的数是按照数值选取，不是按照输入顺序选取，因此需要对每个数字记录出现的次数，存入数组a中，数组的下标就是数字的大小，同时记下数字最大的值，记为ma。然后再进行动归求解；

• 初始化：dp[1]=a[1] (准确来说是1*a[1])
• 转移方程：dp[i]=max(dp[i-1],dp[i-2]+a[i]*i);因为选取x之后，x-1与x+1不可选，因此dp[i]有两种方式，一种是选取i-1,此时i变得不可选，则dp[i]=dp[i-1]；另一种是不选i-1，此时就代表着要选i，dp[i]=dp[i-2]+a[i]*i;最终选取二者中最大的数

最后输出时，输出dp[ma]

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int main()
{//由于不可选取的是数值 x-1与x+1。不是输入顺序，因此要存储每个数字出现的个数 
	long long a[100100],n,ma=-1, dp[100100];
	for(int i=0;i<100100;i++)
	{
		a[i]=0;
		dp[i]=0;
	}
	scanf("%lld",&n);
	for(int i=0;i<n;i++)
	{
		long long tmp;
		scanf("%lld",&tmp);
		a[tmp]++;
		ma=max(ma,tmp);
	}
	dp[1]=a[1];
	for(int i=2;i<ma+3;i++)
	{
		if(dp[i-1]>=dp[i-2]+a[i]*i)
			dp[i]=dp[i-1];
		else
			dp[i]=dp[i-2]+a[i]*i;
	}
	printf("%lld\n",dp[ma]);
	return 0;
 }