题目
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;
}