码迷,mamicode.com
首页 > 其他好文 > 详细

CF #544 div3 E. K Balanced Teams

时间:2020-03-01 12:15:41      阅读:68      评论:0      收藏:0      [点我收藏+]

标签:bre   多少   make   tor   贪心   mod   --   als   class   

 

题意:
有n个学生 要求组成k个小组 每个小组中两两差值不得超过5 可以有学生不被编入组中 求最多可以有多少个学生被编入组中

  n,k<=1e5

   题解:

    考虑dp[i][j],i为前i个学生,j为分了几组的最大人数,不选第i个人,dp[i][j]=dp[i-1][j],

   选第i个人,贪心的选择距离a[i]小于等于5的最远那个点-1,记作pos[i],dp[i][j]=dp[pos[i]][j-1]+i-pos

   dp[i][j]=max(dp[i-1][j],dp[pos[i]][j-1]+i-pos)

   代码

  

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef double db;
typedef pair<int,int> pii;
typedef vector<int> vi;
#define check system("pause")
#define all(x) (x).begin(),(x).end()
#define de(a) cout<<#a<<" = "<<a<<endl
#define dd(a) cout<<#a<<" = "<<a<<" "
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define lowbit(a) ((a)&-(a))
#define INF 0x3f3f3f3f
const ll mod = 1e9+7;
const int N = 5e3+20;
#define dep(i,a,b) for(int i=(a);i>=(b);i--)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define mes(p,b) memset(p,b,sizeof(p))
#define sz(x) int(x.size())
int n,k,dp[N][N],a[N],pos[N];
int main()
{
  ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
  cin>>n>>k;
  rep(i,1,n) cin>>a[i];
  sort(a+1,a+1+n);
  rep(i,1,n){
      dep(j,i,1){
          if(a[i]-a[j]<=5) pos[i]=j-1;
          else break;
      }
  }
  int ans=0;
  dp[1][1]=1;
  rep(i,1,n)
      rep(j,1,k)
        dp[i][j]=max(dp[i-1][j],dp[pos[i]][j-1]+i-pos[i]);
  rep(j,1,k) ans=max(ans,dp[n][j]); 
  cout<<ans;
  return 0;
}

 

CF #544 div3 E. K Balanced Teams

标签:bre   多少   make   tor   贪心   mod   --   als   class   

原文地址:https://www.cnblogs.com/FZUlh/p/12388808.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!