题目描述:
农夫约翰建造了一座有n间牛舍的小屋,牛舍排在一条直线上,第i间牛舍在x
i
的位置,但是约翰的m头牛对小屋很不满意,因此经常互相攻击。约翰为了防止牛之间互相伤害,因此决定把每头牛都放在离其它牛尽可能远的牛舍。也就是要最大化最近的两头牛之间的距离。
牛们并不喜欢这种布局,而且几头牛放在一个隔间里,它们就要发生争斗。为了不让牛互相伤害。John 决定自己给牛分配隔间,使任意两头牛之间的最小距离尽可能的大,那么,这个最大的最小距离是多少呢?
输入样例:
5 3 1 2 8 4 9
输出样例:
3
解题思路及代码:
采用二分法处理
Java代码:(12分,超时了)
import java.util.Arrays; import java.util.Scanner; //对于n个牛舍,有m个牛,需要求牛舍间的最大距离,也就是不断尝试不同的分法,使用二分法就会很快 public class Main { static int home[] = new int[100005]; static int n,m; public static void main(String[] args) { // TODO Auto-generated method stub Scanner scanner = new Scanner(System.in); n = scanner.nextInt(); m = scanner.nextInt(); int ans=0; for(int i=1;i<=n;i++) { home[i]=scanner.nextInt(); } Arrays.sort(home,1,n+1);//java自带的对数组的升序,经过调优的快速排序法,还可以对部分排序,加入相应的下标参数 int left = 1,right = home[n]-home[1];//left是二分查找的最小值,right是最大值 while(left<=right) {//满足此条件一直查找 int mid = (left+right)/2; if(check(mid)) { left = mid+1; ans=mid; }else { right = mid-1; } } System.out.print(ans); } public static boolean check(int d) {//d为牛舍的间隔 int next = home[1]+d;//下一个牛舍的位置,初始化为第二个 int cow = 1;//第一个位置肯定有牛 for(int i=2;i<=n;i++) { if(home[i]>=next) { cow++; next = home[i]+d; } } if(cow>=m) return true; else return false; } }
c++代码:
#include<iostream> #include<algorithm> using namespace std; const int N=1e5+3; int n,m,x[N]; inline bool check(int d){ //以d为答案,看是否正确 int cow=1; int rgt=x[1]+d; for(int i=2;i<=n;i++){ if(x[i]<rgt) continue; //不符合跳过 ++cow; rgt=x[i]+d; //符合计数并更新rgt } return cow>=m; //cow>=m是成立 } int main(){ cin>>n>>m; for(int i=1;i<=n;i++) cin>>x[i]; sort(x+1,x+1+n); //二分查找 int l=0,r=x[n]-x[1]; while(l<=r){ int mid=l+r>>1; if(check(mid)) l=mid+1; else r=mid-1; } cout<<r<<endl; return 0; }