首页 > 代码库 > codeforces455A Boredom (裸DP)

codeforces455A Boredom (裸DP)

题目链接:

http://codeforces.com/problemset/problem/455/A


题意:

从n个数中选出一个数 x,得到x的奖励,但是要去掉所有值为x-1和x+1的数,当所有的数都取光的时候求得到最大的奖励


分析:

很裸的一个DP


dp[i]=max(dp[i-1],dp[i-2]+num[i]*i) 表示取值为i的时候所能得到的最大奖励。


代码如下:

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;

typedef long long LL;
const int maxn = 500010;

int num[maxn];
LL dp[maxn];

int main()
{
    int n;
    while(~scanf("%d",&n)){
        memset(num,0,sizeof(num));
        memset(dp,0,sizeof(dp));
        int mmax=0,x;
        for(int i=0;i<n;i++){
            scanf("%d",&x);
            num[x]++;
            if(x>mmax) mmax=x;
        }
        dp[1]=num[1];
        for(int i=2;i<=mmax;i++)
            dp[i]=max(dp[i-1],dp[i-2]+(LL)i*num[i]);
        printf("%lld\n",dp[mmax]);
    }
    return 0;
}


codeforces455A Boredom (裸DP)