std::ranges::upper_bound() 算法

自 C++20 起

简化

// (1)
constexpr I
  upper_bound( I first, S last, const T& value, Comp comp = {}, Proj proj = {} );

// (2)
constexpr ranges::borrowed_iterator_t<R>
  upper_bound( R&& r, const T& value, Comp comp = {}, Proj proj = {} );

参数类型是泛型的，并具有以下约束

• I - std::forward_iterator

• S - std::sentinel_for<I>

• R - std::ranges::forward_range

• 比较:

  • (1) - indirect_strict_weak_order< const T*, projected<I, Proj>>
  • (2) - indirect_strict_weak_order< const T*, projected<ranges::iterator_t<R>, Proj>>

  （为方便阅读，此处省略了 std:: 命名空间）

• T - （无）

• Proj - (无)

所有重载的 Proj 和 Comp 模板参数具有以下默认类型：std::identity, ranges::less。

详细

// (1)
template<
  std::forward_iterator I,
  std::sentinel_for<I> S,
  class T, class Proj = std::identity,
  std::indirect_strict_weak_order< const T*, std::projected<I, Proj>> Comp = ranges::less
>
constexpr I
  upper_bound( I first, S last, const T& value, Comp comp = {}, Proj proj = {} );

// (2)
template<
  ranges::forward_range R,
  class T,
  class Proj = std::identity,
  std::indirect_strict_weak_order< const T*, std::projected<ranges::iterator_t<R>, Proj>> Comp = ranges::less
>
constexpr ranges::borrowed_iterator_t<R>
  upper_bound( R&& r, const T& value, Comp comp = {}, Proj proj = {} );

• (1) 返回一个迭代器，指向范围 [first; last) 中第一个大于 value 的元素，如果未找到此类元素，则返回 last。

  范围 [first; last) 必须根据表达式 !comp(value, element) 进行分区，即，所有使表达式为 true 的元素必须位于所有使表达式为 false 的元素之前。

  完全排序的范围满足此条件。

• (2) 与 (1) 相同，但使用 r 作为源范围，如同使用 ranges::begin(r) 作为 first 和 ranges::end(r) 作为 last。

本页描述的函数类实体是niebloids。

参数

first last	要检查的部分有序元素范围。
r	要检查的部分有序元素范围。
值	用于比较元素的 `value`。
comp	应用于投影元素的比较谓词。
proj	应用于元素的投影。

返回值

迭代器指向第一个大于 value 的元素，如果未找到此类元素，则返回 last。

复杂度

最多 log2(last - first) + O(1) 次比较和投影应用。

然而，对于不符合 random_access_iterator 模型的迭代器，迭代器增量次数是线性的。

异常

(无)

可能的实现

ranges::upper_bound

struct upper_bound_fn
{
    template<std::forward_iterator I, std::sentinel_for<I> S,
             class T, class Proj = std::identity,
             std::indirect_strict_weak_order<
                 const T*,
                 std::projected<I, Proj>> Comp = ranges::less>
    constexpr I operator()(I first, S last, const T& value,
                           Comp comp = {}, Proj proj = {}) const
    {
        I it;
        std::iter_difference_t<I> count, step;
        count = ranges::distance(first, last);

        while (count > 0)
        {
            it = first;
            step = count / 2;
            ranges::advance(it, step, last);
            if (!comp(value, std::invoke(proj, *it)))
            {
                first = ++it;
                count -= step + 1;
            }
            else
                count = step;
        }
        return first;
    }

    template<ranges::forward_range R, class T, class Proj = std::identity,
             std::indirect_strict_weak_order<
                 const T*,
                 std::projected<ranges::iterator_t<R>, Proj>> Comp = ranges::less>
    constexpr ranges::borrowed_iterator_t<R>
        operator()(R&& r, const T& value, Comp comp = {}, Proj proj = {}) const
    {
        return (*this)(ranges::begin(r), ranges::end(r), value,
                       std::ref(comp), std::ref(proj));
    }
};

inline constexpr upper_bound_fn upper_bound;

示例

Main.cpp

#include <algorithm>
#include <iostream>
#include <iterator>
#include <vector>

int main()
{
    namespace ranges = std::ranges;

    std::vector<int> data {1, 1, 2, 3, 3, 3, 3, 4, 4, 4, 5, 5, 6};

    {
        auto lower = ranges::lower_bound(data.begin(), data.end(), 4);
        auto upper = ranges::upper_bound(data.begin(), data.end(), 4);

        ranges::copy(lower, upper, std::ostream_iterator<int>(std::cout, " "));
        std::cout << '\n';
    }
    {
        auto lower = ranges::lower_bound(data, 3);
        auto upper = ranges::upper_bound(data, 3);

        ranges::copy(lower, upper, std::ostream_iterator<int>(std::cout, " "));
        std::cout << '\n';
    }
}

输出

4 4 4
3 3 3 3