Alex doesn't like boredom. That's why whenever he gets bored, he comes up with games. One long winter evening he came up with a game and decided to play it.
Given a sequence a consisting of n integers. The player can make several steps. In a single step he can choose an element of the sequence (let's denote it ak) and delete it, at that all elements equal to ak + 1 and ak - 1 also must be deleted from the sequence. That step brings ak points to the player.
Alex is a perfectionist, so he decided to get as many points as possible. Help him.
The first line contains integer n (1 ≤ n ≤ 105) that shows how many numbers are in Alex's sequence.
The second line contains n integers a1, a2, ..., an (1 ≤ ai ≤ 105).
Print a single integer — the maximum number of points that Alex can earn.
2 1 2
2
3 1 2 3
4
9 1 2 1 3 2 2 2 2 3
10
Consider the third test example. At first step we need to choose any element equal to 2. After that step our sequence looks like this [2, 2, 2, 2]. Then we do 4 steps, on each step we choose any element equals to 2. In total we earn 10 points.
思路:先按数大小从小到大排序,dp[i][0]表示第i个数不取时的最大得分,dp[i][1]表示第i个数取时的最大得分。
AC代码如下:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
struct node
{ ll val;
ll num;
}box[100010];
ll dp[100010][3],val[100010];
int main()
{ int T,t,n,m,i,j,k,num=0;
scanf("%d",&n);
for(i=1;i<=n;i++)
{ scanf("%d",&k);
val[k]++;
}
for(i=1;i<=100000;i++)
if(val[i]>0)
{ box[++num].val=i;
box[num].num=val[i];
}
dp[1][1]=box[1].val*box[1].num;
dp[1][0]=0;
for(i=2;i<=num;i++)
{ if(box[i].val-box[i-1].val==1)
{ dp[i][0]=max(dp[i-1][0],dp[i-1][1]);
dp[i][1]=dp[i-1][0]+box[i].val*box[i].num;
}
else
{ dp[i][1]=max(dp[i-1][0],dp[i-1][1])+box[i].val*box[i].num;
dp[i][0]=max(dp[i-1][0],dp[i-1][1]);
}
}
printf("%I64d\n",max(dp[num][0],dp[num][1]));
}