codeforce-Balanced Teams(DP)

Balanced Teams

题意

• 给你n个数，将这n个数最多分成k组，但是每组中的数最大差值不超过5，k组中的人数最大

思路

• 先排序，先预处理找出每个数的最左端可以取到的数。
• 再进行dp,dp[i][j]表示前i个人分在j组最多可以放多少人
• 对于第i个人，不取,
  取
  组中不管有多少人，代价都是1，所以第j组竟可能的多放

#include<bits/stdc++.h>
using namespace std;
int a[5005];
int l[5005];
int n,k;
int dp[5005][5005];
int main( )
{
    while(~scanf("%d%d",&n,&k))
    {
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
        }
        sort(a+1,a+n+1);
        for(int i=1; i<=n; i++)
        {
            l[i]=lower_bound(a+1,a+1+n,a[i]-5)-a;//找出左端
        }
        dp[0][0]=0;
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=k; j++)
            {
                dp[i][j]=max(dp[i-1][j],dp[l[i]-1][j-1]+i-l[i]+1);
            }
        }
        printf("%d\n",dp[n][k]);
    }

    return 0;
}