1、lower_bound:查找序列中的第一个出现的值大于等于val的位置
这个序列中可能会有很多重复的元素,也可能所有的元素都相同,为了充分考虑这种边界条件,STL中的lower_bound算法总体上是才用了二分查找的方法,但是由于是查找序列中的第一个出现的值大于等于val的位置,所以算法要在二分查找的基础上做一些细微的改动。
首先是我修改数据结构课本上的二分查找实现的lower_bound算法:
int lower_bound(vector<int> array, int size, int key) { int low = 0, high = size - 1; int mid, pos = 0; //需要pos记录位置 while (low < high) { mid = (low + high) / 2; if (array[mid] < key) //若中位数的值小于key的值,我们要在右边子序列中查找,这时候pos可能是右边子序列的第一个 { low = mid + 1; pos = low; } else { high = mid; //若中位数的值大于等于key,我们要在左边子序列查找,但有可能middle处就是最终位置,所以我们不移动last pos = high; } } return pos; }
2、upper_bound:返回的是最后一个大于等于val的位置,也是有一个新元素val进来时的插入位置。
int upper_bound(vector<int> array, int size, int key) { int low = 0, high = size - 1; int mid, pos = 0; while (low < high) { mid = (low + high) / 2; if (array[mid] <= key) { low = mid + 1; pos = low; } else { high = mid; pos = high; } } return pos; }