拿数问题【动态规化dp】

问题描述

给一个序列，里边有 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;
}