题意:
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 input:
2
1 2
sample output:
2
sample input:
3
1 2 3
sample output:
4
sample input:
9
1 2 1 3 2 2 2 2 3
sample output:
10
对于第三个样例:先选任何一个值为2的元素,最后数组内剩下4个2。然后4次选择2,最终得到10分。
思路:
动态规划,当达到i时,如果之前选择的是i-1,则此时的i不能选则,如果之前选择的是i-2,那么此时i可以选择。构造一个新的数组进行记录,选择数据的每个数作为下标,值为数列中所有此数的总和,这样就可以通过动态规划进行判断,求取输出最大的值。
代码:
#include<iostream>
#include<cmath>
using namespace std;
const int maxn=1e5+10;
int main()
{
long long n;
long long temp;
long long a[maxn];
long long dp[maxn];
long long len=0;
long long te=0;
cin>>n;
for(long long i=1;i<=n;i++)
{
cin>>temp;
a[temp]++;
len=max(len,temp);
te=min(te,temp);
}
for(long long i=te;i<=len;i++)
{
dp[i]=max(dp[i-1],dp[i-2]+a[i]*i);
}
cout<<dp[len];
return 0;
}