二分算法在平时刷题或者参加比赛时经常会用到,但是二分写起来并不容易,所以今天就来总结一下二分的常见写法。
一、二分法
二分查找的基本思想是将n个元素分成大致相等的两部分,取a[n/2]与x做比较,如果x=a[n/2],则找到x,算法中止;如果x<a[n/2],则只要在数组a的左半部分继续搜索x,如果x>a[n/2],则只要在数组a的右半部搜索x
二分常见的分为整数二分和实数二分,其中整数二分较难写,因为会涉及到边界问题导致死循环。
二、二分做题的一般思路
- 确定一个区间,使得目标值一定在区间内
- 找一个性质,满足:
2.1 性质具有二段性
2.2 答案是二段性的分界点
三、实数二分
实数二分较为简单,不涉及边界问题,也不用担心死循环问题,所以一般都能较为轻松的写出。
例题:
代码:
// 时间:2022.07.22 12点49分
// 算法:实数二分
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int main ()
{
double x;
cin>>x;
double l=-10000,r=10000;
while(r-l>1e-8)
{
double mid=(l+r)/2;
if(mid*mid*mid>=x) r=mid;
else l=mid;
}
printf("%lf",l);
}
四、整数二分
整数二分步骤:
- 找一个区间[L,R],使得答案一定在该区间内
- 找一个判断条件,使得该判断条件具有二段性,并且答案一定是该二段性的分界点
- 分析中点mid在该判断条件下是否成立,如果成立,考虑答案在哪个区间;如果不成立,考虑答案在哪个区间
- 如果更新方式写的是R=mid,则不用做任何处理;如果更新方式写的是L=mid,则需要在计算mid时加上1.
第四步就是整数二分一个易错的点,我们来分析一下为什么当L=mid的时候就要变成mid=L+R+1>>1;
我们可以假设这个区间一共就只有两个数,那么此时L=R-1,此时求mid时如果不加1,那么求得的mid= L+R>>1=(R-1+R)>>1=L,那么L=mid=L,将会一直死循环下去,当加1之后,mid=R,此时循环的条件(L< R)将会不成立则退出循环。
4.1 整数二分模板一:
当我们将区间[L, R]划分成[L, mid]和[mid + 1, R]时,其更新操作是r = mid或者l = mid + 1;计算mid时不需要加1。
核心代码:
int bsearch_1(int l, int r)
{
while (l < r)
{
int mid = l + r >> 1;
if (check(mid)) r = mid;
else l = mid + 1;
}
return l;
}
4.2 整数二分模板二:
当我们将区间[L, R]划分成[L, mid - 1]和[mid, R]时,其更新操作是r = mid - 1或者l = mid;此时为了防止死循环,计算mid时需要加1。
核心代码:
int bsearch_2(int l, int r)
{
while (l < r)
{
int mid = l + r + 1 >> 1;
if (check(mid)) l = mid;
else r = mid - 1;
}
return l;
}
4.3 整数二分例题:
完整代码:
// 时间:2022.07.22 11点39分
// 算法:二分
// 终于掌握了整数二分的写法
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=100010;
int n,q;
int a[N];
int main ()
{
cin>>n>>q;
for(int i=0;i<n;i++) scanf("%d",&a[i]);
while(q--){
int x;
scanf("%d",&x);
//二分x的左端点
int l=0,r=n-1;//确定区间范围
while(l<r)
{
int mid=l+r>>1;
if(a[mid]>=x) r=mid;
else l=mid+1;
}
if(a[l]==x)
{
cout<<r<<" ";//输出左端点的位置
//二分x的右端点
r=n-1;//右端点一定在[左端点,n-1]之间
while(l<r)
{
int mid=l+r+1>>1;//防止死循环所以需要加1
if(a[mid]<=x) l=mid;
else r=mid-1;
}
cout<<r<<endl;
}
else
cout<<"-1 -1"<<endl;
}
return 0;
}
4.4 总结:
其实刚开始写二分还是很容易出错的,但是只要能理解二分的核心(算出中点,进行比较,确定范围),其实完全不用背模板,看着题目就能写出来,包括的上面的这个整数二分的例题,其中的判断语句和更新语句都是可以变化的,并不是一成不变的。