Problem D: 集训队组队计划
Time Limit: 1 Sec Memory Limit: 128 MBSubmit: 14 Solved: 8
[ Submit][ Status][ Web Board]
Description
经过1个月的训练,集训工作组决定让大家组队参加组队赛。为了让每一队内部水平尽量平均,集训工作组希望能令每个队能力最强的队员和能力最弱的队员的能力值差值的最大值尽量小。已知每个队由三个人组成,一个人不能加入多个队伍,现在有N个队员,能力分别为a1,a2...an,要从中选出3M个队员组成M支队。请问这M支队伍中,能力差值的最大值最小为多少?
Input
多组数据,第一行有一个整数T,表示有T组数据。(T<=10)
以下每组数据第一行有两个整数N,M。(3<=N<=100000,0<M*3<=N)
第二行N个整数a1,a2......an。(0<=ai<=1e^9)
Output
一个整数,表示最小的最大能力差。
Sample Input
2
6 2
1 3 2 8 7 9
8 2
10 80 90 20 50 60 30 40
Sample Output
2
20
HINT
对样例1:Dmin=max(3-1,9-7);
对样例2:Dmin=max(30-10,60-40)
二分。
n的范围是 3<=N<=100000 自然想到二分法。
最后选出每个队的三个人一定是排过序后连续的三个。
因为有序的三个才能使得三人外其他人与最弱者的差都要比三人中最强者与最弱者差大。
最后的答案差值一定是在0-最大值之间的一个数。所以在0-最大值中二分。构造谓词函数isok(int x)(刘汝佳小白书中的一个说法P(x)).
用isok(int x)删选区间范围。直至最终找到最小答案。
至于isok(int x)如何构造,很简单,从左到右扫一遍就行了。
#include <stdio.h>
#include <algorithm>
#define N 100005
using namespace std;
int a[N];
int n,m;
bool isok(int x)
{
int cnt=0;
for(int i=2;i<n;i++)
if(a[i]-a[i-2]<=x)
i+=2,cnt++;
if(cnt>=m)
return true;
return false;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
sort(a,a+n);
int minn=0,maxn=a[n-1];
while(minn<=maxn)
{
int mid=minn+(maxn-minn)/2;
if(isok(mid))
maxn=mid-1;
else
minn=mid+1;
}
printf("%d\n",minn);
}
return 0;
}