拿数问题2 || 动态规划

题目
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你能得到最大分值。

样例
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

解题思路
拿了 x ，就不能拿 x-1和x+1
所以 Max[i] = max ( Max[i-1] ,Max[i-2]+ i)
其中Max[i]记录【1，i】的最大数和
所以拿Max[i-1]和Max[i-2]+i比较，目的就是为了更新[1,i]的和的最大值

现在，x有多个,所以
Max[i] = max(Max[i-1],Max[i-2]+i* cnt[i]) ,其中cnt[i]是i的个数

代码

#include<iostream>
#include<algorithm>
using namespace std;
const int N = 1e5+100;
#define ll long long
ll a[N];
ll cnt[N];
ll Max[N];
ll n;

int main()
{
	ios::sync_with_stdio(0);
	cin>>n;
	for(ll i=1;i<=n;i++)
	{
		cin>>a[i];
		cnt[a[i]]++;
	}
	sort(a+1,a+n+1);
	for(ll i=a[1];i<=a[n];i++)
	{
		Max[i] = max(Max[i-1],Max[i-2]+i*cnt[i]);
	}
	ll res=-1;
	int ai=1;
	for(ll i=a[1];i<=a[n];i=a[++ai])
	{
		if(res<Max[i]) res = Max[i];
	}
	cout<<res<<endl;
	return 0;
}