【DP】拿数问题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 ≤ 10^5)，表示数字里的元素的个数
第二行包含n个整数a1, a2, …, an (1 ≤ ai ≤ 10^5)

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

思路

有一个很明显的思路就是排序，在输入过程中记录不同的数，并求解相等的数的数目，之后进行排序，在dp的过程中，如果前一个数小于当前数-1，则直接取数，否则，当前数要么取要么不取（可以保证第i-2个数一定小于第i个数）。

代码

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iomanip>
#include<set>
using namespace std;
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define mem(a,s) memset(a,s,sizeof(a))
const int N = 1e5 + 10;
typedef long long ll;
ll dp[N],f[N];
int n;
set<ll> a;
ll sum[N];
int main(){
    //freopen("in.txt","r",stdin);
    scanf("%d", &n);
    rep(i,1,n){
        int x;
        scanf("%d",&x);
        sum[x]++;
        a.insert(x);
    }
    int ii = 0;
    for(auto it:a)
        f[++ii] = it;
    dp[1] = f[1]*sum[f[1]];
    rep(i,2,n){
        if(f[i-1]<f[i]-1)
            dp[i] = dp[i - 1]+f[i]*sum[f[i]];
        else
            dp[i] = max(dp[i - 1], dp[i - 2] + f[i]*sum[f[i]]);
    }
    cout << dp[n];
    return 0;
}