二分答案时,需要有上界、下界、中点。分别命名为left,right,mid。
此题中,要求最大值,因此有重要结论:若mid为解,则最优解一定属于[mid,right](要寻找最小间隔的最大值)。若mid不为解,则最优解一定属于[left,mid)。
AC代码如下
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
void in(vector<int>& a){//输入数据
for(int i=0;i<(int)a.size();i++){
cin>>a[i];
}
}
int Search_L(vector<int>& a){//寻找二分答案的下界
int minn=a[1]-a[0];
for(int i=1;i<(int)a.size();i++){
minn=min(minn,a[i]-a[i-1]);
}
return minn;
}
//关键部分check()
//思路主要为
//mid为当前假定的每头牛间隔的最小值
//先在第一个牛棚放一头牛
//在间隔大于等于mid的时候再放一头牛
//当越过最后一个牛棚的时候
//判断此时放的牛的数量num和真正的牛的数量C作比较
//如果num>=C的话,说明在mid的间隔下能放得下这么多牛,返回true
//此时二分区间应当向右边缩(因为我们要寻找满足条件的间隔的最大值)
//否则的话将区间向左边缩
//当L==R时,此时的值即为答案了,再判断一次后,R指向答案,L指向答案加1
bool check(vector<int> a,int mid,int C){
int num=1,base=a[0];
for(int i=1;i<(int)a.size();i++){
if(a[i]-base>=mid){
base=a[i];
num++;
}
}
return num>=C;
}
int main(){
int N,C;
cin>>N>>C;//定义牛棚数N和牛的数量C
vector<int> a(N);//创建动态数组a
in(a);//通过函数将牛棚的位置输入进去
//为了避免输入数据无序,同时也为了后续check中间方便判断进行排序
sort(a.begin(),a.end());
int L = Search_L(a),R=a[N-1]-a[0],mid;//二分答案的上界与下界还有mid
//Search_L函数是寻找所有牛棚中的最小间隔
//接下来就是二分模板了
while(L<=R){
mid = L+(R-L)/2;
if(check(a,mid,C)) //判断check()是二分答案的关键之处
L=mid+1;
else
R=mid-1;
}
cout<<(R);
}
二分答案相关知识:
二分答案这是非常清楚的一个做题思路。
一般看到所谓的最大值最小化或者最小值最大化一般都是用二分答案进行运算
二分答案的模板很好写,难点主要在检验函数,一般二分答案就是考察这个!
检验函数很重要