C++二分函数之lower_bound()与upper_bound()函数

lower_bound()和upper_bound()的主要用法

一、主要思想

lower_bound()和upper_bound()函数的内部实现是利用 二分查找 ，其时间复杂度为O(logn),其定义在 < algorithm > 中。
需要注意的是，函数必须在有序数组中实现，否则会出错。

二、函数原型

position lower_bound(position_begin,position_end,val ) : 从数组 begin 位置到 end 位置查找第一个大于或等于 val 的数字，也就是数组中第一次出现 val 的位置，返回值为数字的 地址 。
position upper_bound(position_begin,position_end,val ) : 从数组 begin 位置到 end 位置查找第一个严格大于 val 的数字，也就是数组中最后一次出现 val 的位置+1，返回值为数字的 地址 。
当数组中没有 val 时，返回值为数组中最后一个元素的地址+1。
通过返回的地址减去数组的起始地址，也就得到了 val 出现的下标。

三、用法展现

使用 lower_bound() 函数常用来查找元素在数组中第一次出现的位置，而使用 upper_bound() 函数常用来查找元素在数组中最后一次出现的位置。

#include <bits/stdc++.h>
using namespace std;

int main()
{				   //0 1 2 3 4 5 6 7
	vector<int> a = {1,2,2,3,3,3,4,5};
	
	int pos1 = lower_bound(a.begin(),a.end(),3) - a.begin(); //返回数组中第一个大于或等于被查数的值
	int pos2 = upper_bound(a.begin(),a.end(),3) - a.begin(); //返回数组中第一个大于被查数的值
	
	cout << pos1 << " " << pos2 << '\n';//输出3 6
	
	return 0;
}

四、重载

除了可以查找数组中第一个大于或等于 val 的元素之外，我们还可以重载lower_bound()和upper_bound()函数来查找的一个小于或等于 val 的元素。我们主要可以使用c++内部的重载函数 greater < type >() 来实现。
其主要应用在从大到小排序的数组中。

#include <bits/stdc++.h>
using namespace std;

int main()
{				   //0 1 2 3 4 5 6 7
	vector<int> a = {5,4,3,3,3,2,2,1};
	
	int pos1 = lower_bound(a.begin(),a.end(),3,greater<int>()) - a.begin(); //返回数组中第一个小于或等于被查数的值
	int pos2 = upper_bound(a.begin(),a.end(),3,greater<int>()) - a.begin(); //返回数组中第一个小于被查数的值
	
	cout << pos1 << " " << pos2 << '\n';//输出2 5
	
	return 0;
}

好了，lower_bound()和upper_bound()函数讲解就到这里，祝大家学习愉快！