Codeforces1133 E. K Balanced Teams（dp）

题意：

给定n个人的能力值数组a，和一个整数k
要求挑选出一些人，分为最多k组，每组之内的人能力值极差不能超过5
问最多挑出多少人

数据范围：n,k<=5e3

解法：

先对a数组从小到大排序

dp(i，j）表示前i个人分为j组的最大人数

设pos为当前i向左的最大拓展位置，即a[i]-a[pos]<=5

如果不选i，那么dp(i，j)=dp(i-1,j)
如果选i，那么dp(i，j)=dp(pos-1，j-1)+(i-pos+1)
取max更新dp(i，j)即可

ps：
我的dp水平也太捞了吧

code：

#include<bits/stdc++.h>
using namespace std;
const int N=5e3+5;
int d[N][N];
int a[N];
signed main(){
    int n,k;cin>>n>>k;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    sort(a+1,a+1+n);
    int pos=1;//记录当前i向左拓展的最大位置
    for(int i=1;i<=n;i++){
        while(pos<=i&&a[i]-a[pos]>5)pos++;
        for(int j=1;j<=k&&j<=i;j++){
            d[i][j]=max(d[i-1][j],d[pos-1][j-1]+(i-pos+1));
        }
    }
    int ans=0;
    for(int j=1;j<=k&&j<=n;j++){
        ans=max(ans,d[n][j]);
    }
    cout<<ans<<endl;
    return 0;
}

---