农夫约翰搭了一间有N间牛舍的小屋。牛舍排在一条线上,第i号牛舍在xi的位置.但是他的M头牛对小屋很不满意,因此经常互相攻击。约翰为了防止牛之间互相伤害,因此决定把每头牛都放在离其他牛尽可能远的牛舍。也就是要最大化最近的两头牛之间的距离。
输入
第一行输入2个整数N,M
第二行输入N个整数xi
输出
输出一个整数,最大的最近距离
样例
输入
5 3 1 2 8 4 9
输出
3
提示
约定:
2<=N<=100000,2<=M<=N,0<=xi<=10^9
这道题明显就是要用二分,所以我们老样子,先敲个板子。
输入和二分主程序:
#include<bits/stdc++.h>
using namespace std;
int n,m,a[100005];
bool check(int x) {
}
int main() {
cin>>n>>m;
for(int i=1; i<=n; i++) {
scanf("%d",&a[i]);
}
//输入
sort(a+1,a+1+n);
//排序,确保升序
int l=0,r=1000000000,mid,best=-1;
while(l<=r) {
mid=(l+r)/2;
if(check(mid)) {
best=mid;
l=mid+1;
//如果满足,就看看能不能更优
} else {
r=mid-1;
//如果不满足,就降低要求
}
}
//二分
cout<<best;
}
接下来就让我们看看这个check()函数怎么写,我们可以采用贪心:
- 一个一个的给“牛舍间隔”分配距离(必须都满足要求)
- 最后看看能不能全分配完
好了,那按照这个思路,敲出代码:
bool check(int x) {
int j=2,cnt=0;//j表示第几个“牛舍间隔”,cnt表示当前“间隔”已累计了多长
for(int i=2; i<=n&&j<=m; i++) {
cnt+=a[i]-a[i-1];//累积距离
if(cnt>=x) {//如果满足要求了,就跳到下一个
cnt=0;
j++;
}
}
if(j<=m)return false;//如果没分配完,不满足
return true;//否则,满足
}
完整代码:
#include<bits/stdc++.h>
using namespace std;
int n,m,a[100005];
bool check(int x) {
int j=2,cnt=0;
for(int i=2; i<=n&&j<=m; i++) {
cnt+=a[i]-a[i-1];
if(cnt>=x) {
cnt=0;
j++;
}
}
if(j<=m)return false;
return true;
}
int main() {
cin>>n>>m;
for(int i=1; i<=n; i++) {
scanf("%d",&a[i]);
}
sort(a+1,a+1+n);
int l=0,r=1000000000,mid,best=-1;
while(l<=r) {
mid=(l+r)/2;
if(check(mid)) {
best=mid;
l=mid+1;
} else {
r=mid-1;
}
}
cout<<best;
}