拿数问题-动态规划

题目描述

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

解题思路

首先可以明确的是这依然是一个动态规划问题。用f[i]记录第i个数据取得的以A[i]分数为边界的最大值，使用pre[A[i]]记录在i之前的小于A[i]-1的值的序号，使用s[A[i]]记录在第i个元素之前的与A[i]相同值元素的个数。
为了实现此，必须使A[i]数组有序,因此预处理是调用sort函数使得A数组有序。
初始化为f[0]=0,f[1]=A[1].转移方程为f[i]=max(f[pre[A[i]]]+s[A[i]]*A[i],f[i-1]).最后的结果为f[n].

代码

#include<iostream>
#include<algorithm>
using namespace std;

const int maxm=1e5+10;
bool c[maxm];//记录每个分数是否搜索过 
int pre[maxm]