WEEK10作业-C-拿数问题Ⅱ

题意

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

思路

首先我们要知道，能够拿一个数的时候当然要把它全部拿掉分数才会尽可能大。所以我们新建一个新数组a[],a[x]表示值为x的数，a[x]的值为原数列x的个数，我们拿了a[x]，就不能拿a[x-1],a[x+1]，这就转化为了拿数问题。转移方程为：f[i]=max(f[i-1],f[i-2]+a[i]*i)

代码

#include<iostream>
#include<algorithm>
#define maxn 100010
#define ll long long
using namespace std;
ll a[maxn],f[maxn];
int maxx=0;
int main(){
	int n;
	cin>>n;
	for(int i=0;i<=n;i++)
	a[i]=0;
	for(int i=1;i<=n;i++){
		int idx;
		cin>>idx;
		a[idx]++;
		maxx=max(maxx,idx);
	}
	f[1]=a[1];
	for(int i=2;i<=maxx;i++){
		f[i]=max(f[i-1],f[i-2]+a[i]*i);
	}
	cout<<f[maxx]<<endl;
	return 0;
}