Week10 C - 拿数问题 II

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

解题思路
新的拿数要上上个状态找
Code

#include<iostream>
#include<algorithm>
#include<stdio.h>
using namespace std;
int n;
long long a[100001];
long long b[100001];//每种元素出现的次数
long long s[100001];//当前得分
long long ans;
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		scanf("%lld",&a[i]);
		b[a[i]]++;
	}
	sort(a+1,a+1+n);
	for(int i=a[1];i<=a[n];i++){
		s[i]=max(s[i-1],s[i-2]+i*b[i]);//不选=继承上一个点 选=上一个点不选的状态加上当前点
		ans=max(ans,s[i]);
	}
	cout<<ans<<endl;
}