洛谷--P1842--进击的奶牛--二分法

题目描述:

Farmer John建造了一个有N(2<=N<=100,000)个隔间的牛棚,这些隔间分布在一条直线上,坐标是x1,…,xN (0<=xi<=1,000,000,000)。
他的C(2<=C<=N)头牛不满于隔间的位置分布,它们为牛棚里其他的牛的存在而愤怒。为了防止牛之间的互相打斗,Farmer John想把这些牛安置在指定的隔间,所有牛中相邻两头的最近距离越大越好。那么,这个最大的最近距离是多少呢?

输入:

第1行:两个用空格隔开的数字N和C。
第2~N+1行:每行一个整数,表示每个隔间的坐标。

输出:

输出只有一行,即相邻两头牛最大的最近距离。

思路:

看到最大的最近距离,显然是要使用二分法,这道题和跳石头那道题的方法可以说是一模一样,跳石头–二分法

  • 先将牛棚坐标排序,然后二分,假定当前解为mid,judge函数判断mid是否为可行解,若是可行解,就去右边搜索是否有更优的解,若不是可行解,就去左边搜索
  • 最主要的还是judge函数,judge(x),判断解为x是否是可行解,以x为解,判断可以放多少只奶牛,如果可以放的奶牛数大于等于给的奶牛数,那么直接返回true,因为以x为最小距离,可以放的奶牛数大于等于给定的数量,多于的我可以拿走,保证最小的不变就可以,遍历完还没大于给定数量,就返回false,说明这是非法解

代码:

#include <iostream>
#include <algorithm>
using namespace std;
int n,c,l,r,mid,ans;
int a[100005];

bool judge(int x)//x为当前二分的解,判断是否合法
{
    
    
	int sum=1;//可以放的奶牛数量 
	int now=1;//当前所在的位置
	
	for(int i=2;i<=n;i++)//第一个位置放一头牛,从第二个位置开始 
	{
    
    
		if(a[i]-a[now]>=x)//两头奶牛之间距离大于x,就在这个地方放奶牛 
		{
    
    
			sum++;
			now=i; 
		}
		if(sum>=c) //已经放的奶牛数量大于等于有的奶牛,就是可行解 
			return true;
	}
	return false;//遍历结束,奶牛数量还不够,就是非法解 
}

int main()
{
    
    
	cin>>n>>c;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	sort(a+1,a+n+1);
	
	l=1;
	r=a[n]-a[1];
	while(l<=r)
	{
    
    
		mid=(l+r)/2;
		if(judge(mid))//判断当前解是否为可行解 
		{
    
    
			l=mid+1;//如果mid是可行解,那么右边可能会有更优的解 
			ans=mid;
		}else
			r=mid-1;//mid不是可行解,那么就去左边寻找 			
	}
	cout<<ans<<endl; 
	
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_45102820/article/details/114497572