hiho1601最大分数 DP

#1601 : 最大得分

时间限制:10000ms

单点时限:1000ms

内存限制:256MB

描述

小Hi和小Ho在玩一个游戏。给定一个数组A=[A1, A2, ... AN]，小Hi可以指定M个不同的值S1，S2, S3 ... SM，这样他的总得分是 ΣSi × count(Si)。（count(Si)是数组中与Si相等的元素的个数)。

为了增加难度，小Ho要求小Hi选择的S1..SM其中任意两个Si和Sj都满足|Si-Sj| > 1。

你能帮助小Hi算出他最大得分是多少吗？

输入

第一行包含两个整数N和M。

第二行包含N个整数A1, A2, ... AN。

对于30%的数据，1 ≤ M ≤ N ≤ 10

对于100%的数据，1 ≤ M ≤ N ≤ 1000 1 ≤ Ai ≤ 100000

输出

最大得分

样例输入

5 2
1 2 1 2 3

样例输出

        5

 

dp[i][j]表示前i个用了j个的最大值

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<memory>
#include<cstring>
using namespace std;
const int maxn=;
long long  a[maxn],b[maxn],c[maxn];
long long  dp[maxn][maxn],Max;
int main()
{
    long long  n,i,j,k,m,cnt=;
    scanf("%lld%lld",&n,&m);
    for(i=;i<=n;i++) scanf("%lld",&a[i]);
    sort(a+,a+n+);
    for(i=;i<=n;i++){
       if(a[i]!=a[i-]) {
           b[++cnt]=a[i];
           c[cnt]=;
       }
       else  c[cnt]++;
    }
    memset(dp,-,sizeof(dp));
    for(i=;i<=cnt;i++) dp[i][]=;
    dp[][]=b[]*c[];
    Max=dp[][];
    for(i=;i<=cnt;i++)
      for(j=;j<=m;j++){
            dp[i][j]=dp[i-][j];//不用第i个
            if(b[i]-b[i-]>) dp[i][j]=max(dp[i][j],dp[i-][j-]+b[i]*c[i]);//要第i个，前一个也要
            else dp[i][j]=max(dp[i][j],dp[i-][j-]+b[i]*c[i]);//要第i个，而前一个不要
            Max=max(Max,dp[i][j]);
      }
    printf("%lld\n",Max);
    return ;
}