3583. 整数分组（选数最多分成k组，最多能选几个数dp+双指针）

3583. 整数分组（最多分成k组，最多能选几个数）

输入样例1：

5 2
1 2 15 15 15

输出样例1：

5

输入样例2：

6 1
36 4 1 25 9 16

输出样例2：

2

输入样例3：

4 4
1 10 100 1000

输出样例3：

4

在钻石收藏家（确定合适的分组）基础上，如果最多分为k个符合条件的分组，最多可以选出几个数（k个分组由多少数构成）

#include <bits/stdc++.h>
using namespace std;
// #define int long long int
const int N=5e3+5;
int n,T,k;
int a[N];
int dp[N][N];//dp[i][j],在前i个数里选出x个数划分为满足条件的j组 值是x
// 第一维相当于缩小规模，假设只有i(结果是n)个数
// 当有一个满足条件的第j个分组[l,r]时，dp[l-1][j-1]
signed main(){
    cin>>n>>k;//选出的数最多划分为 k组
    for(int i=1;i<=n;i++)cin>>a[i];
    // 1 2 3 1000 101
    sort(a+1,a+n+1);
    int l=1;
    for(int r=1;r<=n;r++){
        while(a[r]-a[l]>5)l++;
        // r-l+1 这个满足条件的区间大小
        for(int j=1;j<=k;j++){
            dp[r][j]=max(dp[r-1][j],dp[l-1][j-1]+r-l+1);
        }
    }
    
    cout<<dp[n][k];
    return 0;
}