WEEK 10 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分。

思路

因为要取到最大分数，所以每当取到一个值x，序列中所有Aj=x都要取，所以我们可以直接将重复的值x进行合并，并记录所有该值的和储存到基本元素类型为pair<int,long long>的数组sum中，并且将合并后的数组sum按值本身（即ans[i].first）设置为升序。
约定ans[i]为取到第i个数时的最大分数，若第二个值恰好是第一个值+1，则第二个值与第一个值只能取一个；否则，第一个值与第二个值皆可取。剩下的ans可使用统一的判断方式进行处理,若当前值与上一个值相差1，则只能二者取一个；否则，两者都可取。所有处理结束后ans数组中的最后一个数即为所求。

代码

#include <iostream>
#include <string>
#include <string.h>
#include <algorithm>
#include <vector>

using namespace std;

typedef long long ll;
const int N = 1e5 + 5;
int num[N], n;
vector<pair<int, ll>> sum;
ll ans[N];

void solve()
{
	sort(num, num + n);
	int r = num[0];
	ll temp = num[0];
	for (int i = 1; i < n; i++)
	{
		if (num[i] == r)
			temp += r;
		else
			sum.push_back(make_pair(r, temp)), r = num[i], temp = num[i];
	}
	sum.push_back(make_pair(r, temp));
	ans[0] = sum[0].second;
	if (sum.size() < 2)
		return;
	if (sum[1].first != sum[0].first + 1)
		ans[1] = ans[0] + sum[1].second;
	else
		ans[1] = max(ans[0], sum[1].second);
	for (int i = 2; i < sum.size(); i++)
	{
		if (sum[i].first != sum[i - 1].first + 1)
			ans[i] = ans[i - 1] + sum[i].second;
		else
			ans[i] = max(sum[i].second + ans[i - 2], ans[i - 1]);
	}
}

int main()
{
	scanf("%d", &n);
	for (int i = 0; i < n; i++)
		scanf("%d", &num[i]);
	solve();
	printf("%lld\n", ans[sum.size() - 1]);
    return 0; 
}