C++ 二分函数 lower_bound, upper_bound

对于一个递增的序列

lower_bound 返回序列中 第一大于等于x 的数 的地址

upper_bound 返回序列中 第一个大于x 的数 的地址

用法：比如查找数组 a ， 在[1 ~ n]中查找第一个大于等于x的数

int *p = lower_bound(a + 1, a + n + 1, x)

lower_bound中第一个参数是查找的起始位置，第二个参数是终止位置，第三个参数是要查找的数

上面返回的是第一个大于等于x的位置

要直接得到查找数的下标可以这么写

int p = lower_bound(a + 1, a + n + 1, x) - a

p 就是数组[1 ~ n] 中第一个大于等于 x 的数的下标

要查找一个递减的序列 需要加入一个参数 greater<int>()

lower_bound(a + 1, a + n + 1, x, greater<int>()> //返回的是第一个 小于等于x 数的位置
upper_bound(a + 1, a + n + 1, x, greater<int>()> //返回的是第一个 小于x 的数 的位置

示例

#include <iostream>
#include <algorithm>

using namespace std;

int main()
{
	int a[10] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
	int p1 = lower_bound(a , a + 10, 5) - a;
	int p2 = upper_bound(a , a + 10, 5) - a;
	cout << "p1 = " << p1 << endl;
	cout << "p2 = " << p2 << endl;
	
	int b[10] = {9, 8, 7, 6, 5, 4, 3, 2, 1, 0};
	int p3 = lower_bound(b, b + 10, 5, greater<int>()) - b;
	int p4 = upper_bound(b, b + 10, 5, greater<int>()) - b;
	
	cout << "p3 = " << p3 << endl;
	cout << "p4 = " << p4 << endl;
	return 0;
}

运行结果