题面:
给一个序列,里边有 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 ≤ 105)
输出一个整数: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分)
思路:
- 给一个序列,里面有n个数,每一步可以拿一个数,如果拿了这个数,那么相邻的两个数就会变得无法选择
- 对于dp转移方程,由于数据范围进行了更改,如果不取i这个数,则值为dp[i-1] ,如果取i这个数,不能取i-1,所以取dp[i-2]+dp[i]*i ,所以转移方程dp[i] = max(dp[i-1], dp[i-2] +a[i]*i)
- 需要注意一下数据范围的问题,虽然a的数据范围在int范围内,但是由于我们求最大分数,所以最终的结果应该用long long 存储
#include<iostream>
#include<cmath>
using namespace std;
const int N=1e5+1;
int main()
{
int n;
long long x;
int a[N];
long long dp[N];
long long length=0;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%lld",&x);
a[x]++;
length=max(length,x);
}
dp[1]=a[1];
for(long long i=2;i<=length;i++)
{
dp[i]=max(dp[i-1],dp[i-2]+a[i]*i);
}
printf("%lld",dp[length]);
return 0;
}