[CF1133E] K Balanced Teams - dp

现有 \(n\) 个人，每个人都有各自的能力值，要你把他们分成 \(k\) 组（每组人数不限），使得每组中任意两个人的能力值之差不超过 \(5\)，问你最多可以把多少人分到组中。

Solution

设 \(f[i][j]\) 表示将前 \(i\) 个人分成 \(j\) 段，且第 \(i\) 个人一定被使用时的最大总人数

\[f[i][j]=\max(f[i-1][j],f[l][j-1]+i-l) \]

其中 \(l=lower\_bound(a[i]-5)-1\)

#include 
using namespace std;

#define int long long
const int N = 5005;

int n,k,a[N],b[N],f[N][N];

signed main() {
    ios::sync_with_stdio(false);
    cin>>n>>k;
    for(int i=1;i<=n;i++) cin>>a[i];
    sort(a+1,a+n+1);
    for(int i=1;i<=n;i++) b[i]=lower_bound(a+1,a+n+1,a[i]-5)-a-1;
    for(int i=1;i<=n;i++) {
        for(int j=1;j<=min(i,k);j++) {
            int l=b[i];
            f[i][j]=max(f[i-1][j],f[l][j-1]+i-l);
        }
    }
    cout<<*max_element(f[n]+1,f[n]+k+1);
}