C++中lower_bound函数和upper_bound函数

时间:2022-09-28 17:42:26
STL 中关于二分查找的函数有三个lower_bound 、upper_bound 、binary_search 。这三个函数都运用于有序区间(当然这也是运用二分查找的前提),下面记录一下这两个函数。

ForwardIter lower_bound(ForwardIter first, ForwardIter last,const _Tp& val)算法返回一个非递减序列[first, last)中的第一个大于等于值val的位置。

ForwardIter upper_bound(ForwardIter first, ForwardIter last, const _Tp& val)算法返回一个非递减序列[first, last)中的第一个大于值val的位置。

lower_bound和upper_bound如下图所示:

C++中lower_bound函数和upper_bound函数

1.lower_bound函数源代码:

01. //这个算法中,first是最终要返回的位置
02. int lower_bound(int *array, int size, int key)
03. {
04. int first = 0, middle;
05. int half, len;
06. len = size;
07.  
08. while(len > 0) {
09. half = len >> 1;
10. middle = first + half;
11. if(array[middle] < key) {    
12. first = middle + 1;         
13. len = len-half-1;       //在右边子序列中查找
14. }
15. else
16. len = half;            //在左边子序列(包含middle)中查找
17. }
18. return first;
19. }

2.upper_bound函数源代码:

01. int upper_bound(int *array, int size, int key)
02. {
03. int len = size-1;
04. int half, middle;
05.  
06. while(len > 0){
07. half = len >> 1;
08. middle = first + half;
09. if(array[middle] > key)     //中位数大于key,在包含last的左半边序列中查找。
10. len = half;
11. else{
12. first = middle + 1;    //中位数小于等于key,在右半边序列中查找。
13. len = len - half - 1;
14. }
15. }
16. return first;
17. }