STL源码分析之二分查找算法

前言

前面我们分析了关于stl_algo.h中的基本算法, 本节也将继续分析该文件中的算法实现, 即二分查找.

stl_algo.h提供了lower_bound和upper_bound两种查找, 前者是找出第一个满足条件的迭代器, 后者则是找出最后一个满足条件的迭代器, 两者结合就能知道该容器中重复的个数.

二分查找分析

lower_bound

ForwardIterator版本

// 版本一
template <class ForwardIterator, class T>
inline ForwardIterator lower_bound(ForwardIterator first, ForwardIterator last, const T& value)
{
   
  return __lower_bound(first, last, value, distance_type(first), iterator_category(first));
}

template <class ForwardIterator, class T, class Distance>
ForwardIterator __lower_bound(ForwardIterator first, ForwardIterator last,
                              const T& value, Distance*,
                              forward_iterator_tag) {
   
  Distance len = 0;
  distance(first, last, len);
  Distance half;
  ForwardIterator middle;

  while (len > 0) {
   
    half = len >> 1;	// 相当于除2
      // middle为区间的起始位置
    middle = first;
      // 设置middle为区间的中间值
    advance(middle, half);
      // 将value值与中间值比较, 即是二分查找, 若中间值小于value, 则继续查找右半部分
    if (*middle < value) {
   
      first = middle;
      ++first;
      len = len - half -