3583. 整数分组

3583. 整数分组

2021/05/30

3583. 整数分组

• 题目
• 视频讲解

给定 nn 个整数 a1,a2,…,ana1,a2,…,an。

现在，请你从中挑选一些数，并将选出的数进行分组。

要求：

1. 选出的数最多划分为 kk 组（至少 11 组）。
2. 同一组内，任意两数之差的绝对值不超过 55。
3. 所选出的数尽可能多。

请问，最多可以选出多少个数进行分组？

输入格式

第一行包含两个整数 nn 和 kk。

第二行包含 nn 个整数 a1,a2,…,ana1,a2,…,an。

输出格式

输出一个整数，表示可以选出的最大整数数量。

数据范围

1≤k≤n≤50001≤k≤n≤5000,
1≤ai≤1091≤ai≤109

输入样例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

AC源码

#include <bits/stdc++.h>
using namespace std;

const int N = 5e3+10;

int a[N];

int f[N][N];
int main()
{
    
    /*
    状态表示：f[i][j]表示只从前i个数中选且分j组的集合！属性：最大能选多少个数！
    状态计算：
        不选i：f[i - 1][j]
        选  i：f[k - 1][j - 1] + i - k + 1 （k为选i的一组的差值小于5最靠左的下标，i - k + 1为选i最后一组的元素个数）

        因此：f[i][j] = max(f[i - 1][j], f[k - 1][j - 1] + i - k + 1)

    最终答案：f[n][m]，m为最大组数，即使最优解不是分为m组，我们也可以将其拆分为m组，最大可选数并不会改变！

    关于如何快速求选i的最后一组差值小于5的最靠左的下标，可以使用双指针扫描：
        1. 该序列为排序好的序列，具有单调性！
        2. i 往前走 k 一定往后走！
*/
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n,m;
    cin>>n>>m;
    for (int i = 1; i <= n; i ++ ){
        cin>>a[i];
    }
    sort(a+1,a+n+1);
    for (int i = 1,k=1; i <= n; i ++ ){
        while(a[i]-a[k]>5) k++;
        for (int j = 1; j <=m; j ++ ){
            f[i][j]=max(f[i-1][j],f[k-1][j-1]+i-k+1);
        }
    }
    cout<<f[n][m]<<endl;
    return 0;
}