Codeforces Round #544 (Div. 3) E. K Balanced Teams（DP）

题目链接


题意：有n个学生，每个学生有ai值，要求把学生分成不超过k组，同时一组内学生的ai值最大不能超过5，求能组队的最大学生数。
思路：排序后，能组队的肯定是连续的数，dp【i】【j】代表前i个学生组成j个组的最大学生数，只要在第二层循环里找到满足条件的pos位置。

#include<bits/stdc++.h>
using namespace std;
const int maxn=5e3+5;
int n,k,ans=0,pos=1,dp[maxn][maxn],a[maxn];
int main()
{
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;++i) scanf("%d",&a[i]);
	sort(a+1,a+1+n);
	for(int i=1;i<=n;++i)
	{
		while(a[i]-a[pos]>5&&pos<=i) pos++;
		for(int j=1;j<=k;++j)
		dp[i][j]=max(dp[i-1][j],dp[pos-1][j-1]+i-pos+1);
	}
	for(int i=1;i<=k;++i) ans=max(ans,dp[n][i]);
	printf("%d\n",ans);
}