upper_bound() 与 lower_bound() 两个函数定义在 <algorithm> 的头文件中, 是一种基于二分查找的方法, 在一个排好序的数组中进行查找目标元素的地址.
简单来说, 如果想在某个排列好的数组中为某一个元素找到一个位置, 并且仍然不破坏数组的排列顺序, 那么我们就可以使用 upper_bound() 与 lower_bound() 找到元素在数组中的地址.
1. 二分查找
二分查找 (Binary search) 是一类优化后的查找算法, 其时间复杂度为 logn (时间复杂度计算log的底数常默认为2).
那么二分查找的原理是什么呢?
我们定义一个数组arr, 一个查找区间 [low, high] 以及一个目标数key, 便可以进行二分查找. 取数组的中间位mid = (low + high) / 2, 比较 key 与 arr[mid]. 若 key < arr[mid], 则 high = mid - 1, 查找在左半区间进行; 若 key > arr[mid], 则 low = mid + 1, 查找在右半区间进行. 直到 key = arr[mid], 则查找成功.
下面是一个比较形象的演示图:
在一个含大量元素的数组里, 二分查找的效率明显高于逐步查找的效率, 而在仅含少量元素的数组里, 二分查找便会显得逊色. 所以, 我们在算法的使用上应"因地制宜".
2. lower_bound() 和 upper_bound() 的用处与用法
现在我们已知, lower_bound() 和 upper_bound() 是一种有效率的查找函数, 那么它会查找哪些东西?
我们定义一个有序数组 A[6] = {1, 1, 3, 3, 5, 6}, 以及一个目标数 x, 令 x = 4.
lower_bound() 二分查找第一个大于或等于 x 的数字, 找到返回该数字的地址, 不存在则返回最后一个数. (值得注意的是, 该函数的查找区间是左开右闭的, 下同)
upper_bound() 二分查找第一个大于 x 的数字, 找到返回该数字的地址, 不存在则返回最后一个数. 下面给一个用法实例.
#include<iostream>
#include<algorithm>
using namespace std;
int k, n = 6;
int A[6] = { 1, 1, 3, 3, 5, 6 };
int main()
{
cin >> k;
cout << k << " 的第一个大于等于它的地址在 " << (lower_bound(A, A + n, k) << endl;
return 0;
}
那么, 要如何才能返回该数字在数组里的位置, 而不是内存地址?
cout << k << "的第一个大于等于它的位置在" << ((lower_bound(A, A + n, k)) - A)<< endl;
将代码改进, 我们会发现, 代码将本该输出地址的 lower_bound 函数与数组A进行了一个相减. 其中A也是一个地址, 两个数组元素地址相减,实际是获取两个元素数组元素的距离. 即可以得出目标元素的位置. 或者可以使用 *lower_bound(A, A+n, k) 获取值.
关于地址的运算, 我后期会单独再写一篇的.
例题:
一串有n个数的数列,数字分别为 a1......an,有次操作,每次操作提供-个数 kj,每次操作时删掉数列中所有比 kj小的数,并求数列还剩几个数,然后接着进行下一次操作(若某次操作结束后数列中无剩余数字,则恢复数列至初始状态,然后再进行下一次操作 )。
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 1000000;
int A[maxn]{}, B[maxn]{};
int n, q, k = 0;
using namespace std;
int main()
{
cin >> n >> q; //输入n, q
for (int a = 0; a < n; a++) //循环为数组A各元素赋a值
{
cin >> A[a];
}
for (int k = 0; k < q; k++) //循环为数组B各元素赋k值
{
cin >> B[k];
}
sort(A, A + n); //快速排序数组A, 按学生血量, 让学生从大到小排序,
int resl, x = 0;
for (int j = 0; j < q; j++) //开始计算结果
{
x = lower_bound(A + x, A + n, B[j]) - A;
resl = n - x;
if (resl == 0)
{
resl = n;
x = 0;
}
cout << resl << " "; //输出结果
}
cout << endl;
return 0;
}
是一个社团的面试题, 也是因为这道题我才学会了 lower_bound 和 upper_bound 函数, 我基本上是碰见问题才学习的那种废物类型.