问题描述
给一个序列,里边有 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
Output
2
Input
3
1 2 3
Output
4
Input
9
1 2 1 3 2 2 2 2 3
Output
10
问题分析
这道题暴力的话肯定会超时,我们考虑dp解决。
设计状态:
首先需要将 A 序列按升序排序,假设结果为 a。
dp[x][0] 表示不取 x 这么大的数的最大值:
dp[x][0] = max( dp[x-1][1], dp[x-1][0] );
dp[x][1] 表示取 x 这么大的数的最大值:
dp[x][1] = dp[x-1][0] + a[i] ( a[i] = x )
答案:
max( dp[a[n-1]][0], dp[a[n-1]][1] )
#include<stdio.h>
#include<algorithm>
using namespace std;
int a[100010];
long long dp[100010][2];
int main(){
int n;
scanf("%d",&n);
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
sort(a,a+n);
dp[0][0]=dp[0][1]=0;
int j=0;
for(int i=1;i<=a[n-1];i++){
long long sum=0;
while(a[j]==i){
sum+=i;
j++;
}
dp[i][0]=max(dp[i-1][1],dp[i-1][0]);
dp[i][1]=dp[i-1][0]+sum;
}
printf("%lld",max(dp[a[n-1]][0],dp[a[n-1]][1]));
return 0;
}