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

Sample Input1

2
    
1 2
   

Sample Output1

2

Sample Input2

3
1 2 3

Sample Output2

4

Sample Input3

9
    
1 2 1 3 2 2 2 2 3
   

Sample Output3

10

思路分析

这道题和课上讲的只有一点点变化。课上讲的是如果拿了a[i]，那么a[i-1]，a[i+1]就不可拿。这道题是如果拿了a[i]，那么a[i]-1,a[i]+1就不可拿。
具体思路：
定义一个数组b[]，b[i]表示以b[i]为拿取的最大分数时最大总分值，方程式：b[a[i]]=max(b[a[i]-1],b[a[i]-2]+a[i])

注意

此题存在一个值有多个的情况，因此定义一个cnt数组统计各个数字出现的次数，方程式改为：b[a[i]]=max(b[a[i]-1],b[a[i]-2]+cnt[a[i]]*a[i])

AC代码

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

const int maxn=1e5+5;
long long n,a[maxn],b[maxn],cnt[maxn],maxx;
int main()
{
 scanf("%lld",&n);
 for(long long i=1;i<=n;i++)
 {
  scanf("%d",&a[i]);
  cnt[a[i]]++;
 }
 sort(a+1,a+n+1);
 for(long long i=a[1];i<=a[n];i++)
 {
  b[i]=max(b[i-1],b[i-2]+i*cnt[i]);
  maxx=max(maxx,b[i]);
 } 
 cout<<maxx<<endl;
 return 0;
}