CF456C Boredom (DP)
CF#260 div2 C. Boredom
Codeforces Round #260
C. Boredom
time limit per test
1 secondmemory limit per test
256 megabytesinput
standard inputoutput
standard outputAlex 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. Input
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). Output
Print a single integer — the maximum number of points that Alex can earn. Sample test(s)
Input
2 Output
2 Input
3 Output
4 Input
9 Output
10 Note
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. |
题意:给出一堆数,一次操作是取某个数,然后等于(它+1)或(它-1)的数全删掉,得到(它)分,求最高得分。
有两种dp方法,一种排序后dp[i][j]表示取到第i个得到的最大分,j表示取或不取;
另一种是我用的,dp[i]表示在1~i的数中取,必取i,得到的最高分,从2扫到100000。
dp[i]=dp[i-1]; if(a[i]>0)dp[i]=max(dp[i],dp[i-2]+1LL*i*a[i]);
注意10^5 * 10^5= 10^10会爆long long,我就是这样掉分飞起的。
1 //#pragma comment(linker, "/STACK:102400000,102400000") 2 #include<cstdio> 3 #include<cmath> 4 #include<iostream> 5 #include<cstring> 6 #include<algorithm> 7 #include<cmath> 8 #include<map> 9 #include<set> 10 #include<stack> 11 #include<queue> 12 using namespace std; 13 #define ll long long 14 #define usint unsigned int 15 #define mz(array) memset(array, 0, sizeof(array)) 16 #define minf(array) memset(array, 0x3f, sizeof(array)) 17 #define REP(i,n) for(i=0;i<(n);i++) 18 #define FOR(i,x,n) for(i=(x);i<=(n);i++) 19 #define RD(x) scanf("%d",&x) 20 #define RD2(x,y) scanf("%d%d",&x,&y) 21 #define RD3(x,y,z) scanf("%d%d%d",&x,&y,&z) 22 #define WN(x) printf("%d\n",x); 23 #define RE freopen("D.in","r",stdin) 24 #define WE freopen("1biao.out","w",stdout) 25 26 const int maxn=111111; 27 int a[maxn]; 28 ll dp[maxn]; 29 30 31 int main(){ 32 ll ma; 33 int n,i,x; 34 while(scanf("%d",&n)!=EOF){ 35 mz(a); 36 for(i=0;i<n;i++){ 37 scanf("%d",&x); 38 a[x]++; 39 } 40 dp[0]=0; 41 dp[1]=a[1]; 42 ma=dp[1]; 43 for(i=2;i<=100000;i++) 44 { 45 dp[i]=dp[i-1]; 46 if(a[i]>0)dp[i]=max(dp[i],dp[i-2]+1LL*i*a[i]); 47 if(dp[i]>ma)ma=dp[i]; 48 } 49 printf("%I64d\n",ma); 50 } 51 return 0; 52 }