简单描述:对于一个有序数组,我们通过check检查某一值,不断缩小范围,找到答案。
二分模板
一、查找精确值模板
以一段简单的代码为例,主要展示的是这个模板.
查找一个有序数组中 5 的下标, 没有返回-1
#include <iostream>
#include <algorithm>
using namespace std;
int a[] = {1, 2, 3, 4, 5, 6, 7};
int f1(int l, int r, int target){
while(l <= r){ //注意这里写等号
int mid = (l + r) >> 1;
if(a[mid] == target) return mid;
if(a[mid] < target){
l = mid + 1;
}
else{
r = mid - 1;
}
}
return -1;
}
int main(){
int target = 5;
int pos = f1(0, 6, target);
cout << pos<< endl;
return 0;
}
二、查找大于/大于等于目标的首元素
更新区间
[l, mid], [mid + 1, r]
#include <iostream>
#include <algorithm>
using namespace std;
int a[] = {1, 2, 3, 4, 5, 6, 7};
int f2(int l, int r, int target){
while(l < r){
int mid = (l + r) >> 1;
if(a[mid] > target) r = mid;
//如果带等号表示返回大于等于target的第一个元素下标
//如果不带等号表示大于target的第一个元素下标
else l = mid + 1;
}
return l;
}
int main(){
int target = 5;
int pos = f2(0, 6, target);
cout << pos<< endl;
return 0;
}
三、查找小于/小于等于目标的首元素
更新区间
[l, mid - 1], [mid, r]
mid属于所划分区间的左半部记得加一
对mid取值的解释
// 当l==r-1时, mid=(l+r)/2=l,l=mid,mid=l…死循环
// 或者, r=l-1, [l,l-1]显然范围出错
#include <iostream>
#include <algorithm>
using namespace std;
int a[] = {1, 2, 3, 4, 5, 6, 7};
int f2(int l, int r, int target){
while(l < r){
int mid = (l + r + 1) >> 1;
if(a[mid] < target) l = mid;
//此处带上=则查找的包含该数本身,否则直接查找第一个小于该数的值
else r = mid - 1;
}
return l;
}
int main(){
int target = 5;
int pos = f2(0, 6, target);
cout << pos<< endl;
return 0;
}
四、实数型的二分
// 实数型二分
double f4(double l, double r)
{
const double eps = 1e-6;
// eps 表示精度, 取决于题目对精度的要求
// 笨方法 for(int i=1; i<=200; i++)
while (r - l > eps)
{
double mid = (l + r) / 2;
if (check(mid)) r = mid;
// check是满足条件的函数
else l = mid;
}
return l;
}
小的细节: 圆周率Π = acos(-1.0)
引入头文件 include <math.h>
include <cmath>
五、两个函数
两个函数均是在已排好序的数组中进行查找的下面讨论升序,降序两种情况
从小到大,升序数组
查找的是> 或>=的值啊
lower_bound
lower_bound(arr, arr+6, num)
:表示从数组的begin位置, 到end - 1( 即[arr, arr + 6)
前闭后开的区间)二分开始查找第一个 >
或 >=
num 的数字, 返回该数字的地址
- begin 得到该数字的下标。
举一个简单例子
#include <bits/stdc++.h>
using namespace std;
int num[] = {1, 2, 4, 7, 15, 80};
int main(){
int pos = lower_bound(num, num + 6, 5) - num;
//输出大于5的第一个元素的下标,3
cout << pos << endl;
return 0;
}
upper_bound
upper_bound(arr, arr+ 6, num)
: 从数组的begin位置开始, 到end - 1, 即[arr, arr + 6)
前闭后开的区间,二分查找第一个 >
num 的数字,返回该数字地址, 不存在返回end。减去begin得到数组中的下标。
#include <bits/stdc++.h>
using namespace std;
int num[] = {1, 2, 4, 7, 15, 80};
int main(){
int pos = upper_bound(num, num + 6, 7) - num;
//输出的是大于 7 的第一个元素下标 4
cout << pos << endl;
return 0;
}
从大到小,降序数组
查找的都是小于或等于num的数字
此处与升序数组差距不大, 仅是把大于
变成小于
。
关于greater <type> ()
的用法,我先不研究,毕竟我是一个连二分模板都没记住的人。。
lower_bound
#include <bits/stdc++.h>
using namespace std;
int num[] = {80, 15, 6, 4, 2, 0};
int main(){
int pos = lower_bound(num, num + 6, 7, greater<int>()) - num;
//输出第一个小于等于7的下标 2.
cout << pos << endl;
return 0;
}
upper_bound
#include <bits/stdc++.h>
using namespace std;
int num[] = {80, 15, 6, 4, 2, 0};
int main(){
int pos = upper_bound(num, num + 6, 6, greater<int>()) - num;
//输出第一个小于6的下标 3.
cout << pos << endl;
return 0;
}
补充lower_bound : 如果返回值 == end, 说明该数不存在