博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
Codeforces Round #544 (Div. 3) E. K Balanced Teams(DP)
阅读量:389 次
发布时间:2019-03-05

本文共 531 字,大约阅读时间需要 1 分钟。

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

#include
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);}

转载地址:http://brewz.baihongyu.com/

你可能感兴趣的文章