题意:
给出 n 个数,选取其中若干个数分别组成至多 k 组,要求每组内最大值与最小值的差值不超过5,求最后被选上的总人数。
题解:
将a[1∼n] 从小到大排序,
f[i][j] 表示到第 i 个数为止,已经组成 j 组,最多可以包含多少个数。
那么,考虑第 i 个数选取与否,如果不选,那么 ,
如果选,那么必然是第 i 个数所在组人数加上前面那些组人数,假设 p 表示距离 a[i]左侧最远的那个位置(满足 ),这里是指p之前的那些组的人数
题目链接:
参考:
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int n, k;
const int maxn = 5000 + 10;
int a[maxn];
int f[maxn][maxn];
int main()
{
cin >> n >> k;
for(int i = 1; i <= n; i++)
cin >> a[i];
sort(a+1, a + n + 1);
int p = 1;
int ans = 1;
memset(f, 0, sizeof(f));
f[1][1] = 1;
for(int i = 2; i <= n; i++)
{
while(p < i && a[i] - a[p] > 5) // p~i为一组
p++;
for(int j = 1; j <= min(k, i); j++) //i可能小于k,也可能大于k(这时就应限制大小)
{
f[i][j] = max(f[i-1][j], (i - p + 1) + f[p-1][j-1]);
ans = max(ans, f[i][j]);
}
}
cout << ans << endl;
}
注:发现从数组从下标1开始输入比较稳妥,可以避免样例1发生p-1=-1的情况