题目:拿数问题||
题意:YJQ 上完第10周的程序设计思维与实践后,想到一个绝妙的主意,他对拿数问题做了一点小修改,使得这道题变成了 拿数问题 II。
给一个序列,里边有 n 个数,每一步能拿走一个数,比如拿第 i 个数, Ai = x,得到相应的分数 x,但拿掉这个 Ai 后,x+1 和 x-1 (如果有 Aj = x+1 或 Aj = x-1 存在) 就会变得不可拿(但是有 Aj = x 的话可以继续拿这个 x)。求最大分数。
输入:
第一行包含一个整数 n (1 ≤ n ≤ 105),表示数字里的元素的个数
第二行包含n个整数a1, a2, …, an (1 ≤ ai ≤ 10^5)
输出:
输出一个整数:n你能得到最大分值。
样例:
解题思路:当拿了x,则x-1和x+1不能拿,那么可以对x进行排序,之后如果a[i]=a[j](其中i=j-1)的话则f[j]=max(f[i-2]+1,f[i])来做,否则用f[j]=f[i]+1来做;
这个题有坑,记得开longlong
代码:
#include<iostream>
#include<algorithm>
#include<map>
using namespace std;
struct node
{
long long cnt;
long long num;
}a[1000005];
long long d[1000005],vis[1000005]={0};
bool cmp(node b,node c)
{
return b.cnt<c.cnt;
}
int main()
{
long long n,x;
cin>>n;
int k=1;
for(int i=1;i<=n;i++)
{
cin>>x;
if(vis[x]==0)
{
a[k].cnt=x;
a[k].num=1;
vis[x]=k;
k++;
}else
{
a[vis[x]].num++;
}
}
sort(a+1,a+k,cmp);
d[1]=a[1].cnt*a[1].num;
d[0]=0;
long long maxn=d[1];
for(int i=2;i<k;i++)
{
if(a[i].cnt-1!=a[i-1].cnt)
{
d[i]=d[i-1]+a[i].cnt*a[i].num;
}else
{
d[i]=max(d[i-1],d[i-2]+a[i].cnt*a[i].num);
}
maxn=max(maxn,d[i]);
}
cout<<maxn;
}