STL中二分查找的函数有三个:lower_bound 、upper_bound 、binary_search 。
- 这三个函数都运用于有序区间(当然这也是运用二分查找的前提)。
- 对比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如下图所示:
1.lower_bound函数源代码:
//这个算法中,first是最终要返回的位置
int lower_bound(int* array, int size, int key)
{
int first = 0, middle;
int half, len;
len = size;
while (len > 0)
{
half = len >> 1;
middle = first + half;
if (array[middle] < key)
{
first = middle + 1;
len = len - half - 1; //在右边子序列中查找
}
else
len = half; //在左边子序列(包含middle)中查找
}
return first;
}
2.upper_bound函数源代码:
int upper_bound(int* array, int size, int key)
{
int len = size - 1;
int half, middle;
while (len > 0)
{
half = len >> 1;
middle = first + half;
if (array[middle] > key) //中位数大于key,在包含last的左半边序列中查找。
len = half;
else
{
first = middle + 1; //中位数小于等于key,在右半边序列中查找。
len = len - half - 1;
}
}
return first;
}